The train test split in the Employee_analysis.ipynb file needs some changes as the test data is too small to make meaningful inferences. Also, there is a need to create .csv files of train and test data so that the cleaned data becomes more accessible.