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

Refactor model class to use an abstract base class #25

Closed ikmckenz closed 4 years ago

ikmckenz commented 4 years ago

Should refactor the models class to use an Abstract Base Class, because the class functionality should mostly remain the same despite underlying model type.

Then make the existing RF model a subclass of the new base class, and any new model implementations (like the Neural Network model) should be subclasses as well.