The Network Optimization

Hooray!! We finally got reasonable results from evaluating the PyTorch Feedforward Neural Network.

ClassifiersF1 ScoreAccuracy ScorePrecision ScoreRecall ScoreROC AUC Score
Random Guessing0.4378250.5005490.3893290.5001290.500473
Logistic Regression0.9318980.9487620.9643990.9015180.940170
Neural Network0.9413300.9550520.9551120.9283780.950201
Random Forest0.9951070.9961770.9908940.9993550.996755
Gaussian Naive Bayes0.8626260.8935300.8654890.8603050.887457

This is really good for us.

We finished up by adjusting hyperparameters.

Documentation and the final presentation were also touched up and results were added in.

Leave a comment

Design a site like this with WordPress.com
Get started