From 024dca0c3fab887a03500dc05841265d17e3d878 Mon Sep 17 00:00:00 2001 From: Hugo LIVET Date: Mon, 12 Feb 2024 17:07:53 +0100 Subject: [PATCH] :sparkles: edit analysis --- src/analysis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/analysis.py b/src/analysis.py index 1d51c11..01865c5 100644 --- a/src/analysis.py +++ b/src/analysis.py @@ -32,7 +32,7 @@ def learning_curve_view(classifier, X_train, y_train, learning_reps): test_scores_mean = np.mean(test_scores, axis=1) plt.plot(train_sizes, train_scores_mean, label='Train') - plt.plot(train_sizes, test_scores_mean, label='Test') + plt.plot(train_sizes, test_scores_mean, label='Validation') plt.xlabel("Training set size") plt.ylabel('Score') plt.title('Learning curve')