Djacon
/

rubert-tiny2-russian-emotion-detection

Text Classification

emotion-detection

emotion-recognition

Inference Endpoints

Model card Files Files and versions Community

Djacon commited on Nov 20, 2023

Commit

04c3bb5

·

1 Parent(s): 2bd3b4a

Update Model

Files changed (2) hide show

config.json +22 -20
pytorch_model.bin +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "cointegrated/rubert-tiny2",
   "architectures": [
     "BertForSequenceClassification"
   ],
@@ -11,28 +11,30 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 312,
   "id2label": {
-    "0": "\u0440\u0430\u0434\u043e\u0441\u0442\u044c",
-    "1": "\u0438\u043d\u0442\u0435\u0440\u0435\u0441",
-    "2": "\u0443\u0434\u0438\u0432\u043b\u0435\u043d\u0438\u0435",
-    "3": "\u043f\u0435\u0447\u0430\u043b\u044c",
-    "4": "\u0433\u043d\u0435\u0432",
-    "5": "\u043e\u0442\u0432\u0440\u0430\u0449\u0435\u043d\u0438\u0435",
-    "6": "\u0441\u0442\u0440\u0430\u0445",
-    "7": "\u0432\u0438\u043d\u0430",
-    "8": "\u043d\u0435\u0439\u0442\u0440\u0430\u043b\u044c\u043d\u043e"
   },
   "initializer_range": 0.02,
   "intermediate_size": 600,
   "label2id": {
-    "\u0432\u0438\u043d\u0430": 7,
-    "\u0433\u043d\u0435\u0432": 4,
-    "\u0438\u043d\u0442\u0435\u0440\u0435\u0441": 1,
-    "\u043d\u0435\u0439\u0442\u0440\u0430\u043b\u044c\u043d\u043e": 8,
-    "\u043e\u0442\u0432\u0440\u0430\u0449\u0435\u043d\u0438\u0435": 5,
-    "\u043f\u0435\u0447\u0430\u043b\u044c": 3,
-    "\u0440\u0430\u0434\u043e\u0441\u0442\u044c": 0,
-    "\u0441\u0442\u0440\u0430\u0445": 6,
-    "\u0443\u0434\u0438\u0432\u043b\u0435\u043d\u0438\u0435": 2
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 2048,
@@ -43,7 +45,7 @@
   "position_embedding_type": "absolute",
   "problem_type": "multi_label_classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.27.4",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 83828

 {
+  "_name_or_path": "emotion_detection_model2",
   "architectures": [
     "BertForSequenceClassification"
   ],
   "hidden_dropout_prob": 0.1,
   "hidden_size": 312,
   "id2label": {
+    "0": "neutral",
+    "1": "joy",
+    "2": "sadness",
+    "3": "anger",
+    "4": "enthusiasm",
+    "5": "surprise",
+    "6": "disgust",
+    "7": "fear",
+    "8": "guilt",
+    "9": "shame"
   },
   "initializer_range": 0.02,
   "intermediate_size": 600,
   "label2id": {
+    "anger": 3,
+    "disgust": 6,
+    "enthusiasm": 4,
+    "fear": 7,
+    "guilt": 8,
+    "joy": 1,
+    "neutral": 0,
+    "sadness": 2,
+    "shame": 9,
+    "surprise": 5
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 2048,
   "position_embedding_type": "absolute",
   "problem_type": "multi_label_classification",
   "torch_dtype": "float32",
+  "transformers_version": "4.27.3",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 83828

pytorch_model.bin CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:294d672e8ad60d9432f90582aed8467791792be10cc80bbeac2284ad5839f9e3
-size 116820959

 version https://git-lfs.github.com/spec/v1
+oid sha256:2686936280a09f2bcc0752a55f5c8a4d0558ab018fdd2554b5e5f14bc092fcb0
+size 116821471