This trains a Distill BERT classifier for different roles of post like emotion, advice, opinion, personal background
The inference is used to obtain the class of new datafiles
The seven codes and no of examples we used to train each code:
Definition | No. of incidents | |
Advice | offers ideas and suggestions actions for coping with challenges | 77 |
Referral | refers to information sources, e.g., books, websites, contacts | 44 |
Facts or situation appraisal | Offers facts or reassesses the situation | 181 |
Personal Experience | Shares personal stories or incidents | 220 |
Opinions | Offers view or judgment formed about something. It is not necessarily based on fact or knowledge | 60 |
Emotional support | Expressing feelings (e.g., fear, joy, despair), listening, and seeking or communicating concern, empathy, or reassurance | 196 |
Others | The comment doesn’t fit any of above | 38 |
To obtain the API for calling our model - please contact k.thaker@pitt.edu
The table below shows the performance of each class:
Support Type |
Support Type Prediction | ||
Precision | Recall | F-score | |
Advice | 0.772 | 0.850 | 0.809 |
Referral | 0.818 | 1.000 | 0.900 |
Facts or situation appraisal | 0.816 | 0.769 | 0.792 |
Personal Experience | 0.948 | 0.873 | 0.909 |
Opinions | 0.807 | 0.807 | 0.807 |
Emotional support | 0.910 | 0.739 | 0.816 |
Average | 0.85 | 0.84 | 0.84 |