Skip to content

Latest commit

 

History

History
 
 

milestone5

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

Final Project Submission

CS109B Project Group 29
Calvin J Chiew, Ji Hua, Tim Hagmann

Final report PDF

https://github.com/greenore/deep-learning-project/blob/master/milestone5/Report.pdf

Screencast on YouTube

https://www.youtube.com/watch?v=oyXUTy3y2-k

Link to cleaned data

https://www.dropbox.com/sh/qt20u1fs2jcf9md/AABEYIxwYyTGG4h7mbrX3Evka?dl=0
(Warning: very large file size!)

Up to date versions of notebooks

README on notebook updates

  • No changes in Milestones 1 and 2
  • Milestone 3: Added parameter tuning via cross-validation for random forest and kNN models. Generated new visualisations showing precision and recall of each model by genre.
  • Milestone 4: Added section exploring predicted probabilities from the CNN. Attempted to tune the optimal predicted probability thresholds for genre classification instead of using output layer.
  • Milestone 5: New notebook showing our attempts to use alternative methods to train the network in a bid to improve its precision and recall. Methods attempted include multi-class classification, binary classification for each genre, and multi-label classification with equal sample sizes per class.