In this project , my objective is to predict software engineers salary based on various features like the country , years of experience and the education level .I performed all steps from choosing the data to be used till making a web app for my model using streamlit . I compared various machine learning model ; linear regression , Decision tree regressor and random forest regressor on the basis of mean squared error . At the end , random forest regressor had the best results .
Here is a screenshot for the models results and another one for boxplot for the the salaries based on countries .
Here is a demo for the machine learning based web app I made using streamlit library
InShot_20220320_230930900.mp4
- Data Cleaning
- Data Encoding
- Machine learning modeling
- web application
- Linear regression
- Decision trees regressor
- Random forest regressor
- Mean squared error
- Mean absolute error
Pandas
Matplotlib
Seaborn
Scikit Learn
Numpy
Streamlit
pickle
pychram
pip install numpy
pip install pandas
pip install seaborn
pip install scikit-learn
pip install matplotlib
pip install streamlit