search

yashsmehta / personality-prediction

Experiments for automated personality detection using Language Models and psycholinguistic features on various famous personality datasets including the Essays dataset (Big-Five)
MIT License
186 stars 51 forks source link

Add an option to display the personality of the essays' users #5

Open Randitech opened 3 years ago

Randitech commented 3 years ago

Hi,

First of all, thank you for your program, which is quite interesting and helpful.

I was wondering if it was possible for you to add an option as to print out (as a csv for instance) the predicted personality for each user of the essays.csv? For now, the model (I'm using the MLP_combined_features.py, but I guess it would be feasible for all of them) only display the accuracy score, and would be of great use to add an option to visualize the predicted personalities. It would also be helpful for people that would like to get the predicted personality of their own dataset's users.

If you could implement this feature, it would be amazing. Thank you!

yashsmehta commented 1 year ago

Hi @Randitech, @edersoncorbari just added functionality to the repository to use our language model to make personality predictions on new text input! Please use: unseen_predictor.py.