Skip to content

The ML task in this assignment is binary classification, stock market prediction, based on daily news dataset.

Notifications You must be signed in to change notification settings

kbmclaren/assn1-CMSC478-ML

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

assn1-CMSC478-ML

Description

Assignmet 1 focused on predicting the larger stock markets movements using daily news headlines.

The Jupyter notebook in this repo is the notebook I submitted to be graded. This notebook was a guided application involving prepocessing a dataset, training and testing binary classifiers (Random Forest and XGBoost), use of confusion matrix, tuning of hyperparameters(GridSearchCV), and plotting and comparing ROC curves.

The end results was that I built two models, a Random Forest binary classifier and and XGBoost binary classifer, then improved the RandomForest model's performance by fine tuning the hyperparameters via a search across the hyperparameter values space (a gridsearch).