UBC-MDS / DSCI522_group_12

MIT License
0 stars 5 forks source link

Predict model script #33

Closed d-sel closed 3 years ago

d-sel commented 3 years ago

23

Created ML script which takes in the pre-processed data, runs it through dummy, random forest and logistic regression. Then, it takes and hyper-parametizes the last two and outputs the score into a df which is output as a csv in a 'results' directory. It also takes the best model and runs it on test and outputs a test score.