Upload trained IMDB sentiment MLP artifacts

README.md

@@ -19,7 +19,7 @@ It uses a TF-IDF text representation followed by a small `scikit-learn` MLP neur
 
 ## Metrics
 
-- Accuracy: 77.00%
+- Accuracy: 82.00%
 - Train samples: 400
 - Test samples: 100
 - Dataset: `imdb_top_500.csv`

metrics.json

@@ -1,10 +1,10 @@
 {
   "repo_id": "ceilf6/imdb-sentiment-mlp",
-  "generated_at": "2026-05-15T11:23:07.005266+00:00",
+  "generated_at": "2026-05-15T11:25:06.574314+00:00",
   "dataset": "data/imdb_top_500.csv",
   "train_size": 400,
   "test_size": 100,
-  "accuracy": 0.77,
+  "accuracy": 0.82,
   "model": {
     "feature_extractor": "TfidfVectorizer",
     "classifier": "MLPClassifier",
@@ -13,6 +13,8 @@
       32
     ],
     "activation": "relu",
+    "solver": "lbfgs",
+    "alpha": 0.001,
     "max_features": 5000,
     "ngram_range": [
       1,
@@ -22,28 +24,28 @@
   },
   "classification_report": {
     "negative": {
-      "precision": 0.7014925373134329,
-      "recall": 0.94,
-      "f1-score": 0.8034188034188035,
+      "precision": 0.8333333333333334,
+      "recall": 0.8,
+      "f1-score": 0.8163265306122449,
       "support": 50.0
     },
     "positive": {
-      "precision": 0.9090909090909091,
-      "recall": 0.6,
-      "f1-score": 0.7228915662650602,
+      "precision": 0.8076923076923077,
+      "recall": 0.84,
+      "f1-score": 0.8235294117647058,
       "support": 50.0
     },
-    "accuracy": 0.77,
+    "accuracy": 0.82,
     "macro avg": {
-      "precision": 0.805291723202171,
-      "recall": 0.77,
-      "f1-score": 0.7631551848419318,
+      "precision": 0.8205128205128205,
+      "recall": 0.8200000000000001,
+      "f1-score": 0.8199279711884754,
       "support": 100.0
     },
     "weighted avg": {
-      "precision": 0.805291723202171,
-      "recall": 0.77,
-      "f1-score": 0.7631551848419318,
+      "precision": 0.8205128205128206,
+      "recall": 0.82,
+      "f1-score": 0.8199279711884754,
       "support": 100.0
     }
   }

model.joblib

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:de2e70e3967580daed24d3de0f5c90efec97cfda949155aac0ded3fc419d3ebd
-size 7931284
+oid sha256:5cfdeef7b56e573ce5f43a1006fa85aef476a69a08ce22b0c5e2a2049ea5e720
+size 5352563

vectorizer.joblib

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3b8a5dcea38f5c07c3a3cbb5f4726b001e3f2eff24b7f85fd04f95c4f0618fb2
+oid sha256:fd6163e5b2c4f76ed06ca3b63ce6e605507335de12600948b0e2194d0be46d70
 size 190443