Stremie commited on
Commit
e112cdd
1 Parent(s): ddc5280
Files changed (2) hide show
  1. config.json +10 -2
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "./ClickbaitDetector/model/roberta-base-clickbait",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
@@ -10,8 +10,16 @@
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
 
 
 
 
13
  "initializer_range": 0.02,
14
  "intermediate_size": 3072,
 
 
 
 
15
  "layer_norm_eps": 1e-05,
16
  "max_position_embeddings": 514,
17
  "model_type": "roberta",
@@ -21,7 +29,7 @@
21
  "position_embedding_type": "absolute",
22
  "problem_type": "single_label_classification",
23
  "torch_dtype": "float32",
24
- "transformers_version": "4.17.0",
25
  "type_vocab_size": 1,
26
  "use_cache": true,
27
  "vocab_size": 50265
 
1
  {
2
+ "_name_or_path": "Stremie/roberta-base-clickbait",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
 
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 768,
13
+ "id2label": {
14
+ "0": "Clickbait",
15
+ "1": "Not Clickbait"
16
+ },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
+ "label2id": {
20
+ "Clickbait": 0,
21
+ "Not Clickbait": 1
22
+ },
23
  "layer_norm_eps": 1e-05,
24
  "max_position_embeddings": 514,
25
  "model_type": "roberta",
 
29
  "position_embedding_type": "absolute",
30
  "problem_type": "single_label_classification",
31
  "torch_dtype": "float32",
32
+ "transformers_version": "4.21.1",
33
  "type_vocab_size": 1,
34
  "use_cache": true,
35
  "vocab_size": 50265
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:689d9c0d45c9d5d230f5d356e8f83d898c57c6ce2c8c9f275f134c327ad0e99f
3
- size 498672237
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8da4046e2a62cf2d186ee32aff6583f95bc9d33dd59dea6bf5339d210f903ef7
3
+ size 498657517