andreas122001 commited on
Commit
c710751
·
1 Parent(s): 15144bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -8
README.md CHANGED
@@ -16,14 +16,14 @@ tags:
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`.
@@ -32,4 +32,5 @@ This is a text classification for detecting machine-generated text based on the
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.
 
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`.
 
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
+
36
  **NOTE**: when not using the attention-head in the predictions, the predictions will be wrong.