empathy_classification ❤️‍🩹

💬 Detect empathy from a sentence

🤗 This is a binary classification model using DistilBert

🦸 Download Model File Drive Link

💭 Dir structure

Notebooks --> contains all the notebooks

src --> contains prediction pipeline change paths in config.py(output_model variable) and main.py(test_data variable)

data --> data for training and testing

run main.py for prediction

👶 Output :

1 : Empathy 🙇

2: Not Empathy 🙅‍♂️