search

ikmckenz / target-pred-py

A simple machine learning model for small-molecule target prediction in Python.
GNU General Public License v3.0
18 stars 8 forks source link

Make neural net hyperparameters controllable #27

Open ikmckenz opened 4 years ago

ikmckenz commented 4 years ago

Make the neural net hyperparameters like epochs, learning rate, momentum, batch size, etc., controllable by class parameters.