A Streamlit web app that predicts whether a person is an introvert or extrovert based on behavioral traits using a Random Forest Classifier.
-
Takes in 4 simple inputs:
- Social event attendance
- Frequency of going outside
- Social media post frequency
- Friends circle size
-
Predicts personality type: Introvert 🪫 or Extrovert 🎉
-
Built with:
- Python 🐍
- Scikit-learn
- Streamlit
- Joblib
-
Source: Kaggle Dataset
-
2900 rows × 8 columns
-
Preprocessing steps:
- Converted categorical variables to binary (Yes/No → 1/0)
- Selected features with positive correlation to target
- Imputed missing values using
SimpleImputer - Used Random Forest for classification
git clone https://github.com/your-username/personality-predictor.git
cd personality-predictorpip install -r requirements.txtstreamlit run app.py├── app.py # Main Streamlit app
├── personality_model.pkl # Trained Random Forest model
├── requirements.txt # Python dependencies
└── README.md # Project documentation
- ✅ Accuracy: 90.3%
- 📈 R² Score: 0.62
- ✔️ Trained using
RandomForestClassifierfrom Scikit-learn
This project is licensed under the MIT License.
Developed by Anik Chand