Skip to content

Commit e5e0919

Browse files
author
SamoraHunter
committed
scale validation for consistency
1 parent b2766ba commit e5e0919

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

ml_grid/pipeline/grid_search_cross_validate.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -186,6 +186,9 @@ def __init__(
186186
self.X_test = pd.DataFrame(
187187
scaler.transform(self.X_test), columns=self.X_test.columns, index=self.X_test.index
188188
)
189+
self.X_test_orig = pd.DataFrame(
190+
scaler.transform(self.X_test_orig), columns=self.X_test_orig.columns, index=self.X_test_orig.index
191+
)
189192

190193
# --- PERFORMANCE FIX for testing ---
191194
# Use a much faster CV strategy when in test_mode.

0 commit comments

Comments
 (0)