This Employee salary prediction uses GradientBoostingClassifier machine learning algorithm to predict the classified predictions which will be above 50000rs or below.
Python Libraries Needed :
- Scikit-learn - for machine learning model building, processing, and evaluation ( GradientBoostingClassifier, train_test_split, LabelEncoder).
- Joblib - for saving and loading models.
- Pandas - for data manipulation and analysis.
- Streamlit - for deploying the model as web application.
Use following commands in cmd to install:
pip install scikit-learn joblib pandas streamlit
TO RUN : run the salary_prediction.ipynb file