This is implementation of Machine Learning graduate course project. The project relies mainly on the numpy, sklearn, pandas, Pytorch, plotly, pickle and matplotlib.
The files should be run in the order:
pre-processing.ipynb
train_*.pynb
GPU is not required
Training for SVM model take quite long time (around 1 hour)
The dataset was obtained from Kaggle: HR Analytics: Job Change of Data Scientists https://www.kaggle.com/arashnic/hr-analytics-job-change-of-data-scientists