hjian42 / automatic-personality-prediction

[AAAI SAP 2020] Modeling Personality with Attentive Networks and Contextual Embeddings
59 stars 19 forks source link

Training model from scratch #4

Open ErikTromp opened 4 years ago

ErikTromp commented 4 years ago

I noticed you specify a lot of different models supported in this repo on which you base your personality classifier. I am looking to use a version (not the default one) of BERT/RoBERTa that I trained myself and base a classifier on top of that with my own dataset.

How would I go by doing this? The code is quite meta and I couldn't find anything related to BERT models being supported other than the readme and the copy of huggingface inside this repo.

dowell666 commented 4 years ago

I noticed you specify a lot of different models supported in this repo on which you base your personality classifier. I am looking to use a version (not the default one) of BERT/RoBERTa that I trained myself and base a classifier on top of that with my own dataset.

How would I go by doing this? The code is quite meta and I couldn't find anything related to BERT models being supported other than the readme and the copy of huggingface inside this repo.