Neural Net predict top doesn't return correct number of entries

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

Neural Net predict top doesn't return correct number of entries #31

Closed ikmckenz closed 4 years ago

ikmckenz commented 4 years ago

When predicting using the neural net model, the predict_top function doesn't listen to n_outputs and always outputs the top 5. It should output n_outputs number of outputs.

ikmckenz commented 4 years ago

Resolved by #32