This project aims to predict the psychological traits of a person on the basis of his/her social media posts. It predicts the following traits:
- EXT - Extraversion
- OPN - Openness
- NEU - Neuorticism
- CON - Conscientiousness
- AGR - Agreeableness
Link to colab notebook This notebook demostrates how our script works.
- Clone the repository
git clone https://github.com/Mainakdeb/e_summit.git
- Navigate to specific Directory
cd e_summit
- Install the requirements from requirements.txt file
pip install -r requirements.txt
- In case you don't want to train the algorithm, download pretrained models from Google drive link
Add desired names in names.txt along with some keywords(some names and keywords already exist in the given file).
Run the command down below
python3 predict_type.py --testcase 7 --inputfile "names.txt"
See the output using:
cat predictions.txt
- Python
- Socialreaper
- SkLearn
- MatplotLib
- Pandas
- GSearch Library
Done | Set Of Tasks |
---|---|
✔️ | Main file and/or other header files if used |
✔️ | The trained model |
✔️ | A Readme with instructions on how to run the code |
✔️ | A brief one page description of the methods used with proper citations |
✔️ | Outcome Report |