minhleduc commited on
Commit
6f796b0
1 Parent(s): 2973aeb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -10,6 +10,8 @@ metrics:
10
  model-index:
11
  - name: toxigen-albert-binary-clsf
12
  results: []
 
 
13
  ---
14
 
15
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,7 +33,7 @@ More information needed
31
 
32
  ## Intended uses & limitations
33
 
34
- More information needed
35
 
36
  ## Training and evaluation data
37
 
@@ -63,4 +65,4 @@ The following hyperparameters were used during training:
63
  - Transformers 4.42.3
64
  - Pytorch 2.1.2
65
  - Datasets 2.20.0
66
- - Tokenizers 0.19.1
 
10
  model-index:
11
  - name: toxigen-albert-binary-clsf
12
  results: []
13
+ datasets:
14
+ - toxigen/toxigen-data
15
  ---
16
 
17
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
33
 
34
  ## Intended uses & limitations
35
 
36
+ Finetuning `albert/albert-base-v1` on the `toxigen/toxigen-data` dataset on the task of binary classfication.
37
 
38
  ## Training and evaluation data
39
 
 
65
  - Transformers 4.42.3
66
  - Pytorch 2.1.2
67
  - Datasets 2.20.0
68
+ - Tokenizers 0.19.1