Johnesss commited on
Commit
04a3f0c
·
verified ·
1 Parent(s): 2bade52

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -2
README.md CHANGED
@@ -1,14 +1,27 @@
1
  ---
2
  library_name: keras
 
 
 
 
 
 
 
3
  ---
4
 
5
  ## Model description
6
 
7
- More information needed
8
 
9
  ## Intended uses & limitations
10
 
11
- More information needed
 
 
 
 
 
 
12
 
13
  ## Training and evaluation data
14
 
 
1
  ---
2
  library_name: keras
3
+ language:
4
+ - en
5
+ pipeline_tag: text-classification
6
+ tags:
7
+ - toxic
8
+ - comment
9
+ - toxic comment
10
  ---
11
 
12
  ## Model description
13
 
14
+ This model used for text classification with toxic and non-toxic labels.
15
 
16
  ## Intended uses & limitations
17
 
18
+ If you want to reuse model, try copy this
19
+ from huggingface_hub import from_pretrained_keras
20
+ reloaded_model = from_pretrained_keras('Johnesss/Toxic-Comment-Classification')
21
+
22
+ y_testing=reloaded_model.predict(x_testing,verbose=1,batch_size=32)
23
+ test_df['Toxic']=['Not Toxic' if x<0.5 else 'Toxic' for x in y_testing]
24
+ test_df[['comment_text','Toxic']].head(20)
25
 
26
  ## Training and evaluation data
27