search

Palashio / libra

Ergonomic machine learning for everyone.
http://libradocs.org/
MIT License
1.92k stars 109 forks source link

Set neural net query to use at most 50 classes for classification #198

Closed jbofill10 closed 4 years ago

jbofill10 commented 4 years ago

Also fixes issue with models not being appended to model_data after fitting

- What I did Changed neural net query to look at unique values to determine whether the problem is a classification one or a regression one. - How I did it Taking value counts of the target due to string and float issues and used the length of that object. - How to verify it Run it

This pull request adds a new feature to Libra. @Palashio, could you please take a look at it?