Aniket Sharma

Machine Learning Zoomcamp Update: Friday, 29 October 2021

Today, I trained four models, Linear Regression, Elastic Net, Random Forest and XGBoost for salary prediction. I also did the hyperparameter tuning for all three models.

Random Forest gave the best performance for this problem.

I, thus, downloaded the XGBoost model for status prediction and Random Forest model for salary prediction as .py files from Google Colab for training the final model.

I created a virtual environment using Pipenv, modified the downloaded scripts, trained the final models and evaluated the performance of the models on the test data.

Estimated Time Taken: 3 hours 25 minutes