arize-ai
/

distilbert_reviews_with_language_drift

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

kikoarizeai commited on Jun 1, 2022

Commit

e4e63ef

•

1 Parent(s): 2a5815c

Update config.json

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -15,9 +15,9 @@
   },
   "initializer_range": 0.02,
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",

   },
   "initializer_range": 0.02,
   "label2id": {
+    "NEGATIVE": 0,
+    "NEUTRAL": 1,
+    "POSITIVE": 2
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",