diff --git a/example_classifier.py b/example_classifier.py index d60a014..bd18de5 100644 --- a/example_classifier.py +++ b/example_classifier.py @@ -37,6 +37,7 @@ accuracy = accuracy_score(y_test, y_pred); print(f"\nAccuracy: {accuracy:.2f}"); +print(f"\nThe model was trained on {X_train.shape[0]} samples and tested on {X_test.shape[0]} samples."); print(f"{X_test.shape[0]} samples were tested, and the model achieved an accuracy of {accuracy:.2f}.")