Skip to content

Commit 7a53821

Browse files
authored
Merge branch 'main' into rope-angles
2 parents 42d0277 + 147dc49 commit 7a53821

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ch06/03_bonus_imdb-classification/train_sklearn_logreg.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def load_dataframes():
1919
return df_train, df_val, df_test
2020

2121

22-
def eval(model, X_train, y_train, X_val, y_val, X_test, y_test):
22+
def eval_model(model, X_train, y_train, X_val, y_val, X_test, y_test):
2323
# Making predictions
2424
y_pred_train = model.predict(X_train)
2525
y_pred_val = model.predict(X_val)
@@ -67,9 +67,9 @@ def eval(model, X_train, y_train, X_val, y_val, X_test, y_test):
6767
dummy_clf.fit(X_train, y_train)
6868

6969
print("Dummy classifier:")
70-
eval(dummy_clf, X_train, y_train, X_val, y_val, X_test, y_test)
70+
eval_model(dummy_clf, X_train, y_train, X_val, y_val, X_test, y_test)
7171

7272
print("\n\nLogistic regression classifier:")
7373
model = LogisticRegression(max_iter=1000)
7474
model.fit(X_train, y_train)
75-
eval(model, X_train, y_train, X_val, y_val, X_test, y_test)
75+
eval_model(model, X_train, y_train, X_val, y_val, X_test, y_test)

0 commit comments

Comments
 (0)