Hugging Face's logo

AIPsy
/
bert-base-client-topic-classification-eng

Text Classification
Transformers
Safetensors
English
bert
PyTorch
multi-class-classification
text-embeddings-inference
Model card Files
xet
 Community
myentity commited on
Commit
5a4beae
·
verified ·
1 Parent(s): 2fe5e38

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +91 -99
README.md CHANGED
@@ -13,54 +13,50 @@ tags:
13
 
14
  This is Bert-base-uncased model fine-tuned for topic classification of therapist remarks in psychotherapeutic contexts. The task is a multi-class classification with the following labels:
15
  ```python
16
- id2label = {0: 'Others',
17
- 1: 'Relationship with a Female Figure',
18
- 2: 'Experiencing and Articulating Emotions',
19
- 3: 'Work and Personal Fulfillment',
20
- 4: 'Relationship with a Male Figure',
21
- 5: 'Meaning-Making and Conceptualization',
22
- 6: 'Education and Personal Development',
23
- 7: 'Experience of Fear and Anxiety',
24
- 8: 'Sexual Relationships and Identity',
25
- 9: 'Expression and Communication Difficulties',
26
- 10: 'Self-Acceptance and Social Relationships',
27
- 11: 'Time Perception and Management',
28
- 12: 'Cyclical Patterns of Experience',
29
- 13: 'Therapy Experience and Reflection',
30
- 14: 'Positive Emotional Experience',
31
- 15: 'Mother-Child Relationship Dynamics',
32
- 16: 'Creativity and Emotional Expression',
33
- 17: 'Humor and Cognitive Reframing',
34
- 18: 'Emotional Disturbance and Regulation',
35
- 19: 'Expressions of Certainty',
36
- 20: 'Guilt and Responsibility',
37
- 21: 'Self-Reflection on Maturity',
38
- 22: 'Sleep-Wake Cycle and Daily Rhythm',
39
- 23: 'Life Orientation and Authenticity',
40
- 24: 'Home and Attachment-Autonomy Conflict',
41
- 25: 'Parenting and Family Responsibility',
42
- 26: 'Internal Conflict and Emotional Regulation',
43
- 27: 'Friendship Dynamics',
44
- 28: 'Reflection and Awareness of Unconsidered Thought',
45
- 29: 'Reading as Cognitive and Reflective Engagement',
46
- 30: 'Processes of Personal Change',
47
- 31: 'Emotional Pain and Hurt',
48
- 32: 'Belief, Faith, and Existential Doubt',
49
- 33: 'Evaluation of Neurotic and Healthy States',
50
- 34: 'Emotional Distress and Negative Evaluation',
51
- 35: 'Experience and Regulation of Anger',
52
- 36: 'Positive Emotional Evaluation',
53
- 37: 'Dream Experience and Symbolic Processing',
54
- 38: 'Father-Child Relationship Dynamics',
55
- 39: 'Problem Framing and Self-Resolution',
56
- 40: 'Personal Learning Experiences',
57
- 41: 'Racial Identity and Intergroup Relations',
58
- 42: 'Healthcare Interactions and Medical Authority',
59
- 43: 'Uncertainty about Correctness',
60
- 44: 'Evaluation of Marriage',
61
- 45: 'Marital Relationship and Communication',
62
- 46: 'Self-Worth and Capacity for Love',
63
- 47: 'Self-Expression Through Appearance'}
64
  ```
65
 
66
 
@@ -99,61 +95,57 @@ Given the broad context of issues discussed in psychotherapeutic sessions, the a
99
  Score metrics of trained model
100
  |Sample|F1 macro|
101
  |:-|:-:|
102
- |Test|0.76|
103
- |Validation|0.76|
104
- |Train|0.99|
105
 
106
  F1 score metrics for test sample across categories
107
  |ID|Topic|Precision|Recall|F1|
108
  |:-:|:-|:-:|:-:|:-:|
109
- | 0 | Others | 0.92 | 0.98 | 0.95 |
110
- | 1 | Relationship with a Female Figure | 0.94 | 0.94 | 0.94 |
111
- | 2 | Experiencing and Articulating Emotions | 0.88 | 0.77 | 0.82 |
112
- | 3 | Work and Personal Fulfillment | 0.71 | 0.72 | 0.72 |
113
- | 4 | Relationship with a Male Figure | 0.90 | 0.92 | 0.91 |
114
- | 5 | Meaning-Making and Conceptualization | 0.69 | 0.80 | 0.74 |
115
- | 6 | Education and Personal Development | 0.84 | 0.85 | 0.85 |
116
- | 7 | Experience of Fear and Anxiety | 0.82 | 0.84 | 0.83 |
117
- | 8 | Sexual Relationships and Identity | 0.79 | 0.75 | 0.77 |
118
- | 9 | Expression and Communication Difficulties | 0.69 | 0.77 | 0.73 |
119
- | 10 | Self-Acceptance and Social Relationships | 0.67 | 0.69 | 0.68 |
120
- | 11 | Time Perception and Management | 0.82 | 0.75 | 0.79 |
121
- | 12 | Cyclical Patterns of Experience | 0.81 | 0.75 | 0.78 |
122
- | 13 | Therapy Experience and Reflection | 0.79 | 0.79 | 0.79 |
123
- | 14 | Positive Emotional Experience | 0.72 | 0.67 | 0.70 |
124
- | 15 | Mother-Child Relationship Dynamics | 0.81 | 0.74 | 0.77 |
125
- | 16 | Creativity and Emotional Expression | 0.78 | 0.72 | 0.75 |
126
- | 17 | Humor and Cognitive Reframing | 0.75 | 0.58 | 0.66 |
127
- | 18 | Emotional Disturbance and Regulation | 0.76 | 0.70 | 0.73 |
128
- | 19 | Expressions of Certainty | 0.86 | 0.90 | 0.88 |
129
- | 20 | Guilt and Responsibility | 0.65 | 0.74 | 0.70 |
130
- | 21 | Self-Reflection on Maturity | 0.64 | 0.65 | 0.64 |
131
- | 22 | Sleep-Wake Cycle and Daily Rhythm | 0.66 | 0.73 | 0.69 |
132
- | 23 | Life Orientation and Authenticity | 0.62 | 0.64 | 0.63 |
133
- | 24 | Home and Attachment-Autonomy Conflict | 0.91 | 0.69 | 0.79 |
134
- | 25 | Parenting and Family Responsibility | 0.83 | 0.73 | 0.78 |
135
- | 26 | Internal Conflict and Emotional Regulation | 0.74 | 0.80 | 0.77 |
136
- | 27 | Friendship Dynamics | 0.64 | 0.66 | 0.65 |
137
- | 28 | Reflection and Awareness of Unconsidered Thought | 0.69 | 0.79 | 0.74 |
138
- | 29 | Reading as Cognitive and Reflective Engagement | 0.74 | 0.81 | 0.78 |
139
- | 30 | Processes of Personal Change | 0.69 | 0.61 | 0.65 |
140
- | 31 | Emotional Pain and Hurt | 0.81 | 0.81 | 0.81 |
141
- | 32 | Belief, Faith, and Existential Doubt | 0.91 | 0.85 | 0.88 |
142
- | 33 | Evaluation of Neurotic and Healthy States | 0.62 | 0.77 | 0.69 |
143
- | 34 | Emotional Distress and Negative Evaluation | 0.66 | 0.69 | 0.67 |
144
- | 35 | Experience and Regulation of Anger | 0.76 | 0.76 | 0.76 |
145
- | 36 | Positive Emotional Evaluation | 0.67 | 0.70 | 0.68 |
146
- | 37 | Dream Experience and Symbolic Processing | 0.83 | 0.88 | 0.86 |
147
- | 38 | Father-Child Relationship Dynamics | 0.83 | 0.71 | 0.77 |
148
- | 39 | Problem Framing and Self-Resolution | 0.71 | 0.56 | 0.62 |
149
- | 40 | Personal Learning Experiences | 0.73 | 0.56 | 0.63 |
150
- | 41 | Racial Identity and Intergroup Relations | 0.73 | 0.92 | 0.81 |
151
- | 42 | Healthcare Interactions and Medical Authority | 0.76 | 0.83 | 0.79 |
152
- | 43 | Uncertainty about Correctness | 0.86 | 0.72 | 0.78 |
153
- | 44 | Evaluation of Marriage | 0.78 | 0.95 | 0.86 |
154
- | 45 | Marital Relationship and Communication | 0.81 | 0.85 | 0.83 |
155
- | 46 | Self-Worth and Capacity for Love | 0.77 | 0.74 | 0.76 |
156
- | 47 | Self-Expression Through Appearance | 0.68 | 0.63 | 0.65 |
157
 
158
 
159
 
 
13
 
14
  This is Bert-base-uncased model fine-tuned for topic classification of therapist remarks in psychotherapeutic contexts. The task is a multi-class classification with the following labels:
15
  ```python
16
+ id2label = {
17
+ 0: 'Relationship with a Female Figure',
18
+ 1: 'Experiencing and Articulating Emotions',
19
+ 2: 'Work and Personal Fulfillment',
20
+ 3: 'Relationship with a Male Figure',
21
+ 4: 'Meaning-Making and Conceptualization',
22
+ 5: 'Education and Personal Development',
23
+ 6: 'Experience of Fear and Anxiety',
24
+ 7: 'Sexual Relationships and Identity',
25
+ 8: 'Expression and Communication Difficulties',
26
+ 9: 'Self-Acceptance and Social Relationships',
27
+ 10: 'Time Perception and Management',
28
+ 11: 'Cyclical Patterns of Experience',
29
+ 12: 'Therapy Experience and Reflection',
30
+ 13: 'Positive Emotional Experience',
31
+ 14: 'Mother-Child Relationship Dynamics',
32
+ 15: 'Creativity and Emotional Expression',
33
+ 16: 'Emotional Disturbance and Regulation',
34
+ 17: 'Expressions of Certainty',
35
+ 18: 'Self-Reflection on Maturity',
36
+ 19: 'Sleep-Wake Cycle and Daily Rhythm',
37
+ 20: 'Life Orientation and Authenticity',
38
+ 21: 'Home and Attachment-Autonomy Conflict',
39
+ 22: 'Parenting and Family Responsibility',
40
+ 23: 'Internal Conflict and Emotional Regulation',
41
+ 24: 'Friendship Dynamics',
42
+ 25: 'Reflection and Awareness of Unconsidered Thought',
43
+ 26: 'Reading as Cognitive and Reflective Engagement',
44
+ 27: 'Processes of Personal Change',
45
+ 28: 'Emotional Pain and Hurt',
46
+ 29: 'Belief, Faith, and Existential Doubt',
47
+ 30: 'Evaluation of Neurotic and Healthy States',
48
+ 31: 'Experience and Regulation of Anger',
49
+ 32: 'Positive Emotional Evaluation',
50
+ 33: 'Dream Experience and Symbolic Processing',
51
+ 34: 'Father-Child Relationship Dynamics',
52
+ 35: 'Problem Framing and Self-Resolution',
53
+ 36: 'Personal Learning Experiences',
54
+ 37: 'Racial Identity and Intergroup Relations',
55
+ 38: 'Healthcare Interactions and Medical Authority',
56
+ 39: 'Uncertainty about Correctness',
57
+ 40: 'Evaluation of Marriage',
58
+ 41: 'Marital Relationship and Communication',
59
+ 42: 'Self-Worth and Capacity for Love'}
 
 
 
 
60
  ```
61
 
62
 
 
95
  Score metrics of trained model
96
  |Sample|F1 macro|
97
  |:-|:-:|
98
+ |Test|0.77|
99
+ |Validation|0.78|
100
+ |Train|0.97|
101
 
102
  F1 score metrics for test sample across categories
103
  |ID|Topic|Precision|Recall|F1|
104
  |:-:|:-|:-:|:-:|:-:|
105
+ | 0 | Relationship with a Female Figure | 0.95 | 0.92 | 0.94 | 232 |
106
+ | 1 | Experiencing and Articulating Emotions | 0.76 | 0.86 | 0.81 | 238 |
107
+ | 2 | Work and Personal Fulfillment | 0.74 | 0.76 | 0.75 | 234 |
108
+ | 3 | Relationship with a Male Figure | 0.90 | 0.92 | 0.91 | 150 |
109
+ | 4 | Meaning-Making and Conceptualization | 0.76 | 0.83 | 0.79 | 277 |
110
+ | 5 | Education and Personal Development | 0.87 | 0.86 | 0.86 | 83 |
111
+ | 6 | Experience of Fear and Anxiety | 0.85 | 0.72 | 0.78 | 80 |
112
+ | 7 | Sexual Relationships and Identity | 0.86 | 0.76 | 0.81 | 66 |
113
+ | 8 | Expression and Communication Difficulties | 0.73 | 0.80 | 0.76 | 142 |
114
+ | 9 | Self-Acceptance and Social Relationships | 0.71 | 0.79 | 0.75 | 220 |
115
+ | 10 | Time Perception and Management | 0.81 | 0.74 | 0.77 | 95 |
116
+ | 11 | Cyclical Patterns of Experience | 0.71 | 0.75 | 0.73 | 175 |
117
+ | 12 | Therapy Experience and Reflection | 0.78 | 0.82 | 0.80 | 57 |
118
+ | 13 | Positive Emotional Experience | 0.76 | 0.73 | 0.75 | 78 |
119
+ | 14 | Mother-Child Relationship Dynamics | 0.83 | 0.79 | 0.81 | 61 |
120
+ | 15 | Creativity and Emotional Expression | 0.76 | 0.76 | 0.76 | 37 |
121
+ | 16 | Emotional Disturbance and Regulation | 0.75 | 0.75 | 0.75 | 193 |
122
+ | 17 | Expressions of Certainty | 0.90 | 0.94 | 0.92 | 207 |
123
+ | 18 | Self-Reflection on Maturity | 0.70 | 0.84 | 0.77 | 45 |
124
+ | 19 | Sleep-Wake Cycle and Daily Rhythm | 0.70 | 0.68 | 0.69 | 41 |
125
+ | 20 | Life Orientation and Authenticity | 0.77 | 0.72 | 0.74 | 82 |
126
+ | 21 | Home and Attachment-Autonomy Conflict | 0.86 | 0.81 | 0.84 | 47 |
127
+ | 22 | Parenting and Family Responsibility | 0.79 | 0.85 | 0.82 | 53 |
128
+ | 23 | Internal Conflict and Emotional Regulation | 0.81 | 0.67 | 0.73 | 39 |
129
+ | 24 | Friendship Dynamics | 0.85 | 0.69 | 0.77 | 59 |
130
+ | 25 | Reflection and Awareness of Unconsidered Thought | 0.64 | 0.74 | 0.69 | 94 |
131
+ | 26 | Reading as Cognitive and Reflective Engagement | 0.72 | 0.89 | 0.79 | 35 |
132
+ | 27 | Processes of Personal Change | 0.78 | 0.54 | 0.64 | 39 |
133
+ | 28 | Emotional Pain and Hurt | 1.00 | 0.53 | 0.69 | 32 |
134
+ | 29 | Belief, Faith, and Existential Doubt | 0.81 | 0.78 | 0.79 | 32 |
135
+ | 30 | Evaluation of Neurotic and Healthy States | 0.64 | 0.56 | 0.60 | 32 |
136
+ | 31 | Experience and Regulation of Anger | 0.86 | 0.73 | 0.79 | 33 |
137
+ | 32 | Positive Emotional Evaluation | 0.76 | 0.67 | 0.71 | 48 |
138
+ | 33 | Dream Experience and Symbolic Processing | 0.93 | 0.74 | 0.82 | 19 |
139
+ | 34 | Father-Child Relationship Dynamics | 0.80 | 0.80 | 0.80 | 30 |
140
+ | 35 | Problem Framing and Self-Resolution | 0.86 | 0.73 | 0.79 | 41 |
141
+ | 36 | Personal Learning Experiences | 0.64 | 0.42 | 0.51 | 33 |
142
+ | 37 | Racial Identity and Intergroup Relations | 0.88 | 0.94 | 0.91 | 16 |
143
+ | 38 | Healthcare Interactions and Medical Authority | 0.86 | 0.72 | 0.78 | 25 |
144
+ | 39 | Uncertainty about Correctness | 0.89 | 0.72 | 0.80 | 120 |
145
+ | 40 | Evaluation of Marriage | 0.95 | 0.83 | 0.88 | 23 |
146
+ | 41 | Marital Relationship and Communication | 0.83 | 0.71 | 0.77 | 21 |
147
+ | 42 | Self-Worth and Capacity for Love | 0.79 | 0.86 | 0.83 | 22 |
148
+
 
 
 
 
149
 
150
 
151