|
|
--- |
|
|
base_model: |
|
|
- google-bert/bert-base-cased |
|
|
- google-bert/bert-large-cased |
|
|
- FacebookAI/roberta-base |
|
|
- FacebookAI/roberta-large |
|
|
--- |
|
|
This repository includes the models for a research project focused on Personality Trait Prediction using four models: BERT base, BERT large, RoBERTa base and RoBERTa large. The objective is to investigate how well these models predict personality traits and identify correlations among them based on user-generated text collected from Reddit (arXiv:2004.04460v3). |
|
|
Unlike the models used in the multiple-models approach, the models for the single-model approach are not included separately, as the final RoBERTa base model was directly used for evaluation and comparison in the single-model setting.\ |
|
|
The full paper of this project can be found at the following link: https://www.mdpi.com/2078-2489/16/5/418 |
|
|
|
|
|
The models are organised into the following modules:\ |
|
|
bertbase_best_model : Best BERT base model saved during the fine-tuning process\ |
|
|
bertbase_final_model : Final BERT base model saved after the fine-tuning process\ |
|
|
bertlarge_best_model : Best BERT large model saved during the fine-tuning process\ |
|
|
bertlarge_final_model : Final BERT large model saved after the fine-tuning process\ |
|
|
robertabase_final_model : Final RoBERTa base model saved after the fine-tuning process\ |
|
|
robertalarge_best_model : Best RoBERTa large model saved during the fine-tuning process\ |
|
|
robertalarge_final_model : Final RoBERTa large model saved after the fine-tuning process\ |
|
|
r_a_best_model : Best RoBERTa base model for Agreeableness saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_a_final_model : Best RoBERTa base model for Agreeableness saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_c_best_model : Best RoBERTa base model for Conscientiousness saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_c_final_model : Best RoBERTa base model for Conscientiousness saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_e_best_model : Best RoBERTa base model for Extraversion saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_e_final_model : Best RoBERTa base model for Extraversion saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_n_best_model : Best RoBERTa base model for Neuroticism saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_n_final_model : Best RoBERTa base model for Neuroticism saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_o_best_model : Best RoBERTa base model for Openness saved during the fine-tuning process for the multiple-models approach\ |
|
|
r_o_final_model : Best RoBERTa base model for Openness saved during the fine-tuning process for the multiple-models approach\ |
|
|
\ |
|
|
*Note: The final model for RoBERTa base was also the best model; therefore, no separate best checkpoint was saved. |
