gsar78 commited on
Commit
ec8425f
·
verified ·
1 Parent(s): e75791b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -6
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "gsar78/HellenicSentimentAI",
3
  "architectures": [
4
  "CustomModel"
5
  ],
@@ -27,15 +27,16 @@
27
  "max_position_embeddings": 514,
28
  "model_type": "xlm-roberta",
29
  "num_attention_heads": 12,
 
30
  "num_hidden_layers": 12,
 
31
  "output_past": true,
32
  "pad_token_id": 1,
33
  "position_embedding_type": "absolute",
34
- "problem_type": "multi_label_classification",
35
  "torch_dtype": "float32",
36
- "transformers_version": "4.42.3",
37
  "type_vocab_size": 1,
38
  "use_cache": true,
39
- "vocab_size": 250002,
40
- "num_emotion_labels": 18 // Custom parameter
41
- }
 
1
  {
2
+ "_name_or_path": "/teamspace/studios/this_studio/saved_model",
3
  "architectures": [
4
  "CustomModel"
5
  ],
 
27
  "max_position_embeddings": 514,
28
  "model_type": "xlm-roberta",
29
  "num_attention_heads": 12,
30
+ "num_emotion_labels": 18,
31
  "num_hidden_layers": 12,
32
+ "num_sentiment_labels": 3,
33
  "output_past": true,
34
  "pad_token_id": 1,
35
  "position_embedding_type": "absolute",
36
+ "problem_type": "single_label_classification",
37
  "torch_dtype": "float32",
38
+ "transformers_version": "4.42.4",
39
  "type_vocab_size": 1,
40
  "use_cache": true,
41
+ "vocab_size": 250002
42
+ }