Mr-Vicky-01 commited on
Commit
c75287b
·
verified ·
1 Parent(s): 8e6967f

Upload DistilBertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +18 -22
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,37 +1,33 @@
1
  {
2
- "_name_or_path": "finetuned_TFPF_model",
 
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
- "hidden_act": "gelu",
11
- "hidden_dropout_prob": 0.1,
12
- "hidden_size": 768,
13
  "id2label": {
14
  "0": "FALSE_POSITIVE",
15
  "1": "TRUE_POSITIVE"
16
  },
17
  "initializer_range": 0.02,
18
- "intermediate_size": 3072,
19
  "label2id": {
20
  "FALSE_POSITIVE": 0,
21
  "TRUE_POSITIVE": 1
22
  },
23
- "layer_norm_eps": 1e-05,
24
- "max_position_embeddings": 514,
25
- "model_type": "roberta",
26
- "num_attention_heads": 12,
27
- "num_hidden_layers": 12,
28
- "output_past": true,
29
- "pad_token_id": 1,
30
- "position_embedding_type": "absolute",
31
  "problem_type": "single_label_classification",
 
 
 
 
32
  "torch_dtype": "float32",
33
- "transformers_version": "4.48.3",
34
- "type_vocab_size": 1,
35
- "use_cache": true,
36
- "vocab_size": 50265
37
  }
 
1
  {
2
+ "_name_or_path": "distilbert/distilbert-base-uncased",
3
+ "activation": "gelu",
4
  "architectures": [
5
+ "DistilBertForSequenceClassification"
6
  ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "hidden_dim": 3072,
 
 
 
11
  "id2label": {
12
  "0": "FALSE_POSITIVE",
13
  "1": "TRUE_POSITIVE"
14
  },
15
  "initializer_range": 0.02,
 
16
  "label2id": {
17
  "FALSE_POSITIVE": 0,
18
  "TRUE_POSITIVE": 1
19
  },
20
+ "max_position_embeddings": 512,
21
+ "model_type": "distilbert",
22
+ "n_heads": 12,
23
+ "n_layers": 6,
24
+ "pad_token_id": 0,
 
 
 
25
  "problem_type": "single_label_classification",
26
+ "qa_dropout": 0.1,
27
+ "seq_classif_dropout": 0.2,
28
+ "sinusoidal_pos_embds": false,
29
+ "tie_weights_": true,
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.49.0",
32
+ "vocab_size": 30522
 
 
33
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04b592e6c555527c5124077269ad8dd27313a072b76534e3a5bcf94c4d34ddc6
3
- size 498612824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:072646928d97df19e9f0ee0a29eaf8d8aa17a3b63c4a5ffa0f99cfb0b1d95fcc
3
+ size 267832560