From 36e4a0c71d7c5bd2b72ac9b6260b9bf98024a0d8 Mon Sep 17 00:00:00 2001 From: Anil Sorathiya Date: Tue, 11 Mar 2025 13:00:06 +0000 Subject: [PATCH] Add number of test and training records columns in the result --- validmind/tests/model_validation/sklearn/OverfitDiagnosis.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/validmind/tests/model_validation/sklearn/OverfitDiagnosis.py b/validmind/tests/model_validation/sklearn/OverfitDiagnosis.py index 759024718..0ef87f5f2 100644 --- a/validmind/tests/model_validation/sklearn/OverfitDiagnosis.py +++ b/validmind/tests/model_validation/sklearn/OverfitDiagnosis.py @@ -73,6 +73,7 @@ def _prepare_results( columns={"shape": "training records", f"{metric}": f"training {metric}"}, inplace=True, ) + results["test records"] = results_test["shape"] results[f"test {metric}"] = results_test[metric] # Adjust gap calculation based on metric directionality @@ -292,7 +293,8 @@ def OverfitDiagnosis( { "Feature": feature_column, "Slice": row["slice"], - "Number of Records": row["training records"], + "Number of Training Records": row["training records"], + "Number of Test Records": row["test records"], f"Training {metric.upper()}": row[f"training {metric}"], f"Test {metric.upper()}": row[f"test {metric}"], "Gap": row["gap"],