Skip to content

This trains a Distill BERT classifier for different roles of post like emotion, advice, opinion, personal background

License

Notifications You must be signed in to change notification settings

HELPeR-codes/Post-Support-Type-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Post Support Type Prediction

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

About

This trains a Distill BERT classifier for different roles of post like emotion, advice, opinion, personal background

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published