andreas122001 commited on
Commit
98b2da4
1 Parent(s): 1b40104

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -6
README.md CHANGED
@@ -16,14 +16,20 @@ tags:
16
  - mgt-detection
17
  - ai-detection
18
  metrics:
19
- - accuracy
20
- - precision
21
- - recall
22
- - f1
 
 
 
 
23
  ---
24
 
25
  The hosted inference here, for some reason, does not work. It seems like this is because the hosted inference ignores the attention-mask when doing predictions. It works with pipelines however, and when including attention-mask in the model, e.g. `**encodings`.
26
 
27
- [will update readme later]
 
28
 
29
- This is a text classification for detecting machine-generated text based on the {_name_or_path}
 
 
16
  - mgt-detection
17
  - ai-detection
18
  metrics:
19
+ - type: accuracy
20
+ value: 0.973
21
+ - type: precision
22
+ value: 1.000
23
+ - type: recall
24
+ value: 0.945
25
+ - type: f1
26
+ value: 0.972
27
  ---
28
 
29
  The hosted inference here, for some reason, does not work. It seems like this is because the hosted inference ignores the attention-mask when doing predictions. It works with pipelines however, and when including attention-mask in the model, e.g. `**encodings`.
30
 
31
+ This is a text classification for detecting machine-generated text based on the bloomz-560m by BigScience (see https://huggingface.co/bigscience/bloomz-560m).
32
+ The model is fine-tuned on generations by GPT-2.
33
 
34
+ **NOTE**: the hosted inference does not work. To do inference, please download the model to perform inference.
35
+ **NOTE**: when not using the attention-head in the predictions, the predictions will be wrong.