jasmeeetsingh commited on
Commit
205a718
1 Parent(s): 4a2eb6d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +17 -35
config.json CHANGED
@@ -1,37 +1,19 @@
1
  {
2
- "_name_or_path": "jasmeeetsingh/twitter-depression-classification-sentiment140",
3
- "architectures": [
4
- "RobertaForSequenceClassification"
5
- ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "bos_token_id": 0,
8
- "classifier_dropout": null,
9
- "eos_token_id": 2,
10
- "gradient_checkpointing": false,
11
- "hidden_act": "gelu",
12
- "hidden_dropout_prob": 0.1,
13
- "hidden_size": 768,
14
- "id2label": {
15
- "0": "not depression",
16
- "1": "moderate"
17
- },
18
- "initializer_range": 0.02,
19
- "intermediate_size": 3072,
20
- "label2id": {
21
- "not depression": 0,
22
- "moderate": 1
23
- },
24
- "layer_norm_eps": 1e-05,
25
- "max_position_embeddings": 514,
26
- "model_type": "roberta",
27
- "num_attention_heads": 12,
28
- "num_hidden_layers": 12,
29
- "pad_token_id": 1,
30
- "position_embedding_type": "absolute",
31
- "problem_type": "single_label_classification",
32
- "torch_dtype": "float32",
33
- "transformers_version": "4.18.0",
34
- "type_vocab_size": 1,
35
- "use_cache": true,
36
- "vocab_size": 50265
37
  }
 
1
  {
2
+ "architectures": [
3
+ "TweetDepressionClassifier"
4
+ ],
5
+ "model_type": "roberta",
6
+ "tokenizer_name": "roberta-base",
7
+ "transformers_version": "4.13.0",
8
+ "num_labels": 2,
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 3072,
13
+ "layer_norm_eps": 1e-05,
14
+ "max_position_embeddings": 512,
15
+ "num_attention_heads": 12,
16
+ "num_hidden_layers": 12,
17
+ "type_vocab_size": 1,
18
+ "vocab_size": 50265
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  }