RyanDDD commited on
Commit
5e377fb
·
verified ·
1 Parent(s): e3f70ac

Add pipeline_tag to enable Inference API

Browse files
Files changed (1) hide show
  1. README.md +19 -0
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"