system HF staff commited on
Commit
cd50cab
1 Parent(s): 8ecff19

Commit From AutoTrain

Browse files
.gitattributes CHANGED
@@ -26,3 +26,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
26
  *.zip filter=lfs diff=lfs merge=lfs -text
27
  *.zstandard filter=lfs diff=lfs merge=lfs -text
28
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
26
  *.zip filter=lfs diff=lfs merge=lfs -text
27
  *.zstandard filter=lfs diff=lfs merge=lfs -text
28
  *tfevents* filter=lfs diff=lfs merge=lfs -text
29
+ *.tar.gz filter=lfs diff=lfs merge=lfs -text
30
+ *.pkl filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags: autotrain
3
+ language: en
4
+ widget:
5
+ - text: "I love AutoTrain 🤗"
6
+ datasets:
7
+ - rabiaqayyum/autotrain-data-mental-health-analysis
8
+ co2_eq_emissions: 313.3534743349287
9
+ ---
10
+
11
+ # Model Trained Using AutoTrain
12
+
13
+ - Problem type: Multi-class Classification
14
+ - Model ID: 752423172
15
+ - CO2 Emissions (in grams): 313.3534743349287
16
+
17
+ ## Validation Metrics
18
+
19
+ - Loss: 0.6064515113830566
20
+ - Accuracy: 0.805171240644137
21
+ - Macro F1: 0.7253473044054398
22
+ - Micro F1: 0.805171240644137
23
+ - Weighted F1: 0.7970679970423672
24
+ - Macro Precision: 0.7477679873153633
25
+ - Micro Precision: 0.805171240644137
26
+ - Weighted Precision: 0.7966263131173029
27
+ - Macro Recall: 0.7143231260991618
28
+ - Micro Recall: 0.805171240644137
29
+ - Weighted Recall: 0.805171240644137
30
+
31
+
32
+ ## Usage
33
+
34
+ You can use cURL to access this model:
35
+
36
+ ```
37
+ $ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoTrain"}' https://api-inference.huggingface.co/models/rabiaqayyum/autotrain-mental-health-analysis-752423172
38
+ ```
39
+
40
+ Or Python API:
41
+
42
+ ```
43
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
44
+
45
+ model = AutoModelForSequenceClassification.from_pretrained("rabiaqayyum/autotrain-mental-health-analysis-752423172", use_auth_token=True)
46
+
47
+ tokenizer = AutoTokenizer.from_pretrained("rabiaqayyum/autotrain-mental-health-analysis-752423172", use_auth_token=True)
48
+
49
+ inputs = tokenizer("I love AutoTrain", return_tensors="pt")
50
+
51
+ outputs = model(**inputs)
52
+ ```
config.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "AutoTrain",
3
+ "_num_labels": 7,
4
+ "architectures": [
5
+ "RobertaForSequenceClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.1,
8
+ "bos_token_id": 0,
9
+ "classifier_dropout": null,
10
+ "eos_token_id": 2,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "Anxiety",
16
+ "1": "BPD",
17
+ "2": "autism",
18
+ "3": "bipolar",
19
+ "4": "depression",
20
+ "5": "mentalhealth",
21
+ "6": "schizophrenia"
22
+ },
23
+ "initializer_range": 0.02,
24
+ "intermediate_size": 3072,
25
+ "label2id": {
26
+ "Anxiety": 0,
27
+ "BPD": 1,
28
+ "autism": 2,
29
+ "bipolar": 3,
30
+ "depression": 4,
31
+ "mentalhealth": 5,
32
+ "schizophrenia": 6
33
+ },
34
+ "layer_norm_eps": 1e-05,
35
+ "max_length": 256,
36
+ "max_position_embeddings": 514,
37
+ "model_type": "roberta",
38
+ "num_attention_heads": 12,
39
+ "num_hidden_layers": 12,
40
+ "pad_token_id": 1,
41
+ "padding": "max_length",
42
+ "position_embedding_type": "absolute",
43
+ "problem_type": "single_label_classification",
44
+ "torch_dtype": "float32",
45
+ "transformers_version": "4.15.0",
46
+ "type_vocab_size": 1,
47
+ "use_cache": true,
48
+ "vocab_size": 50265
49
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fee5fc9157471d66eb7c47910bbb017b2465d919fab692a88aa84d8a9d94856f
3
+ size 498689453
sample_input.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:076bffaba201a64fe941419f0246ec53b0f1ed7299864f563c80567c7f62a179
3
+ size 5110
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
1
+ {"bos_token": "<s>", "eos_token": "</s>", "unk_token": "<unk>", "sep_token": "</s>", "pad_token": "<pad>", "cls_token": "<s>", "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": false}}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
1
+ {"unk_token": "<unk>", "bos_token": "<s>", "eos_token": "</s>", "add_prefix_space": false, "errors": "replace", "sep_token": "</s>", "cls_token": "<s>", "pad_token": "<pad>", "mask_token": "<mask>", "trim_offsets": true, "model_max_length": 512, "special_tokens_map_file": null, "name_or_path": "AutoTrain", "tokenizer_class": "RobertaTokenizer"}
vocab.json ADDED
The diff for this file is too large to render. See raw diff