books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename eval method (#832)

Browse Source
This commit is contained in:
Sebastian Raschka
2025-09-15 21:47:20 -05:00
committed by GitHub
parent 8add26cbe9
commit 147dc49ab5
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ch06/03_bonus_imdb-classification/train_sklearn_logreg.py
View File

@@ -19,7 +19,7 @@ def load_dataframes():
return df_train, df_val, df_test
def eval(model, X_train, y_train, X_val, y_val, X_test, y_test):
def eval_model(model, X_train, y_train, X_val, y_val, X_test, y_test):
# Making predictions
y_pred_train = model.predict(X_train)
y_pred_val = model.predict(X_val)
@@ -67,9 +67,9 @@ if __name__ == "__main__":
dummy_clf.fit(X_train, y_train)
print("Dummy classifier:")
eval(dummy_clf, X_train, y_train, X_val, y_val, X_test, y_test)
eval_model(dummy_clf, X_train, y_train, X_val, y_val, X_test, y_test)
print("\n\nLogistic regression classifier:")
model = LogisticRegression(max_iter=1000)
model.fit(X_train, y_train)
eval(model, X_train, y_train, X_val, y_val, X_test, y_test)
eval_model(model, X_train, y_train, X_val, y_val, X_test, y_test)