abhishek HF staff commited on
Commit
aaefa95
1 Parent(s): 11e49b6

Commit From AutoTrain

Browse files
.gitattributes CHANGED
@@ -25,3 +25,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar.gz filter=lfs diff=lfs merge=lfs -text
29
+ *.pkl filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags: autotrain
3
+ language: unk
4
+ widget:
5
+ - text: "I love AutoTrain 🤗"
6
+ datasets:
7
+ - vlsb/autotrain-data-security-text-classification-albert
8
+ co2_eq_emissions: 3.670416179055797
9
+ ---
10
+
11
+ # Model Trained Using AutoTrain
12
+
13
+ - Problem type: Binary Classification
14
+ - Model ID: 688320769
15
+ - CO2 Emissions (in grams): 3.670416179055797
16
+
17
+ ## Validation Metrics
18
+
19
+ - Loss: 0.3046899139881134
20
+ - Accuracy: 0.8826530612244898
21
+ - Precision: 0.9181818181818182
22
+ - Recall: 0.8782608695652174
23
+ - AUC: 0.9423510466988727
24
+ - F1: 0.8977777777777778
25
+
26
+ ## Usage
27
+
28
+ You can use cURL to access this model:
29
+
30
+ ```
31
+ $ 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/vlsb/autotrain-security-text-classification-albert-688320769
32
+ ```
33
+
34
+ Or Python API:
35
+
36
+ ```
37
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
38
+
39
+ model = AutoModelForSequenceClassification.from_pretrained("vlsb/autotrain-security-text-classification-albert-688320769", use_auth_token=True)
40
+
41
+ tokenizer = AutoTokenizer.from_pretrained("vlsb/autotrain-security-text-classification-albert-688320769", use_auth_token=True)
42
+
43
+ inputs = tokenizer("I love AutoTrain", return_tensors="pt")
44
+
45
+ outputs = model(**inputs)
46
+ ```
config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "AutoTrain",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "AlbertForSequenceClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0,
8
+ "bos_token_id": 2,
9
+ "classifier_dropout_prob": 0.1,
10
+ "down_scale_factor": 1,
11
+ "embedding_size": 128,
12
+ "eos_token_id": 3,
13
+ "gap_size": 0,
14
+ "hidden_act": "gelu_new",
15
+ "hidden_dropout_prob": 0,
16
+ "hidden_size": 768,
17
+ "id2label": {
18
+ "0": "irrelevant",
19
+ "1": "relevant"
20
+ },
21
+ "initializer_range": 0.02,
22
+ "inner_group_num": 1,
23
+ "intermediate_size": 3072,
24
+ "label2id": {
25
+ "irrelevant": 0,
26
+ "relevant": 1
27
+ },
28
+ "layer_norm_eps": 1e-12,
29
+ "max_length": 512,
30
+ "max_position_embeddings": 512,
31
+ "model_type": "albert",
32
+ "net_structure_type": 0,
33
+ "num_attention_heads": 12,
34
+ "num_hidden_groups": 1,
35
+ "num_hidden_layers": 12,
36
+ "num_memory_blocks": 0,
37
+ "pad_token_id": 0,
38
+ "padding": "max_length",
39
+ "position_embedding_type": "absolute",
40
+ "problem_type": "single_label_classification",
41
+ "torch_dtype": "float32",
42
+ "transformers_version": "4.15.0",
43
+ "type_vocab_size": 2,
44
+ "vocab_size": 30000
45
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d592fd43674b648564d783d038305c72ec3e12f2c22eadca7f915cb7318f8fed
3
+ size 46755537
sample_input.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5cc78ffa562e55ea2202f83634c00710246f9da8b4574e73c38aec7bdec29ae8
3
+ size 13606
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": "[CLS]", "eos_token": "[SEP]", "unk_token": "<unk>", "sep_token": "[SEP]", "pad_token": "<pad>", "cls_token": "[CLS]", "mask_token": {"content": "[MASK]", "single_word": false, "lstrip": true, "rstrip": false, "normalized": false}}
spiece.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fefb02b667a6c5c2fe27602d28e5fb3428f66ab89c7d6f388e7c8d44a02d0336
3
+ size 760289
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"do_lower_case": true, "remove_space": true, "keep_accents": false, "bos_token": "[CLS]", "eos_token": "[SEP]", "unk_token": "<unk>", "sep_token": "[SEP]", "pad_token": "<pad>", "cls_token": "[CLS]", "mask_token": {"content": "[MASK]", "single_word": false, "lstrip": true, "rstrip": false, "normalized": false, "__type": "AddedToken"}, "model_max_length": 512, "special_tokens_map_file": null, "name_or_path": "AutoTrain", "tokenizer_class": "AlbertTokenizer"}