518_project Code base

logistic_regression.ipynb

This file contains the code for the logistic regression model. It contains the code for the following:

• Data visualization
• Model training
• Model evaluation

random_forest.ipynb

This file contains the code for the random forest model. It contains the code for the following:

• Data visualization
• Model training
• Model evaluation

prepocessing folder

Contains code to calculate the correlation tables that are used in choosing the features for the models. Also contain many initial explorations of the data.