GautamR commited on
Commit
bbeecf0
·
1 Parent(s): d008c1a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -9,6 +9,7 @@ pipeline_tag: text-classification
9
  Chakshu/conversation_terminator_classifier
10
  This model is a fine-tuned version of google/mobilebert-uncased on an unknown dataset. It achieves the following results on the evaluation set:
11
 
12
- Train Loss: 0.0364
13
- Train Binary Accuracy: 0.9915
14
- Epoch: 8
 
 
9
  Chakshu/conversation_terminator_classifier
10
  This model is a fine-tuned version of google/mobilebert-uncased on an unknown dataset. It achieves the following results on the evaluation set:
11
 
12
+ The labels are :
13
+
14
+ - 0 : 'other general conversation'
15
+ - 1 : 'conversation starter'