Add pipeline_tag to enable Inference API
Browse files
README.md
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
---
|
| 2 |
language: en
|
| 3 |
license: mit
|
|
|
|
| 4 |
pipeline_tag: text-classification
|
| 5 |
tags:
|
| 6 |
- text-classification
|
|
@@ -9,6 +10,8 @@ tags:
|
|
| 9 |
- mental-health
|
| 10 |
- counseling
|
| 11 |
- psychology
|
|
|
|
|
|
|
| 12 |
datasets:
|
| 13 |
- AnnoMI
|
| 14 |
metrics:
|
|
@@ -16,6 +19,22 @@ metrics:
|
|
| 16 |
- f1
|
| 17 |
- precision
|
| 18 |
- recall
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
widget:
|
| 20 |
- text: "I really want to quit smoking."
|
| 21 |
example_title: "Change Talk"
|
|
|
|
| 1 |
---
|
| 2 |
language: en
|
| 3 |
license: mit
|
| 4 |
+
library_name: transformers
|
| 5 |
pipeline_tag: text-classification
|
| 6 |
tags:
|
| 7 |
- text-classification
|
|
|
|
| 10 |
- mental-health
|
| 11 |
- counseling
|
| 12 |
- psychology
|
| 13 |
+
- transformers
|
| 14 |
+
- pytorch
|
| 15 |
datasets:
|
| 16 |
- AnnoMI
|
| 17 |
metrics:
|
|
|
|
| 19 |
- f1
|
| 20 |
- precision
|
| 21 |
- recall
|
| 22 |
+
model-index:
|
| 23 |
+
- name: bert-motivational-interviewing
|
| 24 |
+
results:
|
| 25 |
+
- task:
|
| 26 |
+
type: text-classification
|
| 27 |
+
name: Text Classification
|
| 28 |
+
dataset:
|
| 29 |
+
name: AnnoMI
|
| 30 |
+
type: AnnoMI
|
| 31 |
+
metrics:
|
| 32 |
+
- type: accuracy
|
| 33 |
+
value: 0.701
|
| 34 |
+
name: Accuracy
|
| 35 |
+
- type: f1
|
| 36 |
+
value: 0.579
|
| 37 |
+
name: F1 Score (macro)
|
| 38 |
widget:
|
| 39 |
- text: "I really want to quit smoking."
|
| 40 |
example_title: "Change Talk"
|