Skip to content

Yasaman-Mah/STAD68Proj

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

STAD68 Project

Dependencies

  • python 3
  • tensorflow
  • numpy
  • matplotlib
  • scikit-learn

Preparing data

Training

Neural Net with Cross-entropy

  • run FCNN_softmax.py

Neural Net with SVM layer

  • run FCNN_SVM.py

Random Forest

  • run random_forest.py. Note this script does both training and testing

Performance on test set

  • Please run random_forest.py first, it calculates precision-recall scores that will be used in NN_perform.py to plot precision recall curves. It also prints aucpr for the random forest along with confidence intervals.
  • For the neural networks run NN_perform.py, it prints aucpr for the two models, and plot precision recall curve for all models.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages