BaderLab / saber

Saber is a deep-learning based tool for information extraction in the biomedical domain. Pull requests are welcome! Note: this is a work in progress. Many things are broken, and the codebase is not stable.
https://baderlab.github.io/saber/
MIT License
102 stars 17 forks source link

Metrics class should behave more like Keras's default History Callback #149

Open JohnGiorgi opened 5 years ago

JohnGiorgi commented 5 years ago

The Metrics class is a bit of a mess. It should behave more like a Keras History callback, and should be the returned object from a call to saber.train().

This would make hyperparm tuning much easier. E.g. a simple script could be written to collect the average performance of some metric or loss from the Metrics object returned by saber.train(), and tune against it.