Proto007 / MBTI-predictor

A webapp using a linear SVM model to predict mbti personality types given a paragraph.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MBTI-predictor

A webapp using a linear SVM model to predict mbti personality types given a paragraph. The model was trained using MBTI-500 dataset. The model was 84% accurate on the test set.

Demo

Video Walkthrough

This is how the main page looks like:

main

This is how the prediction page looks like:

prediction

Responsive design made to work on mobile as well

Video Walkthrough

Running Locally

  1. Clone the repository: git clone https://github.com/Proto007/MBTI-predictor/
  2. Create and activate a virtualenvironment: https://docs.python.org/3/library/venv.html#creating-virtual-environments
  3. Go to the directory of the project and install necessary dependencies:
cd MBTI-predictor
pip install -r requirements.txt
  1. Go to the django project directory and make database migrations:
cd mbti_predictor
python manage.py makemigrations
python manage.py migrate
  1. Run webapp in localhost: python manage.py runserver

Tools used

Frontend: Django-template, Bootstrap CSS

Backend: Django-Rest-Framework

ML framework: scikit-learn

About

A webapp using a linear SVM model to predict mbti personality types given a paragraph.

License:MIT License


Languages

Language:Python 56.7%Language:HTML 43.3%