Skip to content

Commit 0bc0486

Browse files
Update traintest.py
1 parent 6709f68 commit 0bc0486

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/spotPython/sklearn/traintest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,11 @@ def evaluate_hold_out(model, fun_control):
3535
test_size=0.25,
3636
stratify=train_df[target_column],
3737
)
38+
# scaler fit_transform(X_train)
3839
model.fit(X_train, y_train)
3940
# convert to numpy array, see https://github.com/scikit-learn/scikit-learn/pull/26772
4041
X_test = np.array(X_test)
42+
# scaler transform(X_test)
4143
y_test = np.array(y_test)
4244
if fun_control["predict_proba"]:
4345
df_preds = model.predict_proba(X_test)

0 commit comments

Comments
 (0)