Metrics for hyperparameter selection
Reference issue
Closes #3 (closed)
What does this implement or fix?
- Adds
scoring
parameter tomachine_learn
- Adds
criterion
parameter todeep_learn
- Changes selection of hyperparameters to be based on loss instead of accuracy for
deep_learn
- Fixes legacy PyTorch code by replacing
outputs.data
byoutputs
(cf. PyTorch release notes)
Edited by Johann Benerradi