search

AdityaKane2001 / ACL_WASSA

Code for WASSA'22 shared task.
Apache License 2.0
7 stars 3 forks source link

Added dataset, dataloader, BERT for emotion classification, etc. #8

Closed AdityaKane2001 closed 2 years ago

AdityaKane2001 commented 2 years ago

@sahilkhose @neeraja1504

File structure is as follows:

ACL_WASSA                                   
├─ models                                   
│  ├─ essaytoallbert.py  
│  ├─ essaytoemotionbert.py                     
│  ├─ essaytoemotionempathydistressbert.py  
│  └─ __init__.py                                     
├─ dataloader.py                            
├─ LICENSE                                  
├─ README.md                                
├─ train.py                                 
├─ utils.py                                 
└─ __init__.py

Files in models/ contain a single class for a particular variant of the model.

dataloader.py contains torch.utils.data.Dataset subclass for the training dataset.

/cc @shantypat /auto Closes #3