ONNX
English
gabrielchua commited on
Commit
505c86b
1 Parent(s): d9707d3

Update README to correct model info

Browse files
Files changed (1) hide show
  1. README.md +18 -18
README.md CHANGED
@@ -1,19 +1,19 @@
1
- ---
2
- license: other
3
- license_name: govtech-singapore
4
- license_link: LICENSE
5
- datasets:
6
- - gabrielchua/off-topic
7
- language:
8
- - en
9
- metrics:
10
- - roc_auc
11
- - f1
12
- - precision
13
- - recall
14
- base_model:
15
- - cross-encoder/stsb-roberta-base
16
- ---
17
 
18
  # Off-Topic Classification Model
19
 
@@ -21,8 +21,8 @@ This model leverages a fine-tuned **Cross Encoder STSB Roberta Base** to perform
21
 
22
  ## Model Highlights
23
 
24
- - **Base Model**: [`stsb-roberta-base`](https://huggingface.co/sentence-transformers/stsb-roberta-base)
25
- - **Maximum Context Length**: 1024 tokens
26
  - **Task**: Binary classification (on-topic/off-topic)
27
 
28
  ## Performance
 
1
+ ---
2
+ license: other
3
+ license_name: govtech-singapore
4
+ license_link: LICENSE
5
+ datasets:
6
+ - gabrielchua/off-topic
7
+ language:
8
+ - en
9
+ metrics:
10
+ - roc_auc
11
+ - f1
12
+ - precision
13
+ - recall
14
+ base_model:
15
+ - cross-encoder/stsb-roberta-base
16
+ ---
17
 
18
  # Off-Topic Classification Model
19
 
 
21
 
22
  ## Model Highlights
23
 
24
+ - **Base Model**: [`stsb-roberta-base`](https://huggingface.co/cross-encoder/stsb-roberta-base)
25
+ - **Maximum Context Length**: 514 tokens
26
  - **Task**: Binary classification (on-topic/off-topic)
27
 
28
  ## Performance