andythetechnerd03 commited on
Commit
ed6c68c
·
verified ·
1 Parent(s): acbd97b

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. README.md +5 -5
  2. config.json +9 -1
README.md CHANGED
@@ -1,9 +1,11 @@
1
  ---
 
 
2
  license: mit
 
 
3
  datasets:
4
  - andythetechnerd03/AI-human-text
5
- language:
6
- - en
7
  metrics:
8
  - accuracy
9
  - f1
@@ -11,6 +13,4 @@ metrics:
11
  - precision
12
  - roc_auc
13
  pipeline_tag: text-classification
14
- tags:
15
- - code
16
- ---
 
1
  ---
2
+ language:
3
+ - en
4
  license: mit
5
+ tags:
6
+ - code
7
  datasets:
8
  - andythetechnerd03/AI-human-text
 
 
9
  metrics:
10
  - accuracy
11
  - f1
 
13
  - precision
14
  - roc_auc
15
  pipeline_tag: text-classification
16
+ ---
 
 
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/bert_uncased_L-2_H-128_A-2",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
@@ -8,8 +8,16 @@
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 128,
 
 
 
 
11
  "initializer_range": 0.02,
12
  "intermediate_size": 512,
 
 
 
 
13
  "layer_norm_eps": 1e-12,
14
  "max_position_embeddings": 512,
15
  "model_type": "bert",
 
1
  {
2
+ "_name_or_path": "andythetechnerd03/BERT-tiny_AI-Human",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 128,
11
+ "id2label": {
12
+ "0": "Human",
13
+ "1": "AI"
14
+ },
15
  "initializer_range": 0.02,
16
  "intermediate_size": 512,
17
+ "label2id": {
18
+ "AI": 0,
19
+ "Human": 1
20
+ },
21
  "layer_norm_eps": 1e-12,
22
  "max_position_embeddings": 512,
23
  "model_type": "bert",