A beginner-friendly machine learning project to predict student burnout risk using daily lifestyle inputs.
- Synthetic Data Generation: Custom script to create a balanced dataset.
- Machine Learning Model: Random Forest Classifier trained to categorize burnout risk.
- Interactive UI: Polished Streamlit application with sliders, prediction cards, and personalized advice.
- Visual Feedback: Color-coded risk levels (Green: Low, Yellow: Medium, Red: High).
data_generation.py: Script to generatestudent_burnout_data.csv.train_model.py: Script to train the model and savemodel.pklandscaler.pkl.app.py: Main Streamlit application file.requirements.txt: Python dependencies.
Open your terminal and run:
pip install -r requirements.txtRun these scripts in order to set up the model:
python data_generation.py
python train_model.pyStart the Streamlit dashboard:
streamlit run app.pyThe system takes the following inputs:
- Study Hours: Mental load per day.
- Sleep Hours: Recovery factor.
- Screen Time: Digital fatigue factor.
- Stress Level: Subjective emotional state.
- Assignments: Weekly academic pressure.
MIT