Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Experiment with different classifier models #18

Open
ikmckenz opened this issue Aug 25, 2020 · 1 comment
Open

Experiment with different classifier models #18

ikmckenz opened this issue Aug 25, 2020 · 1 comment
Labels
enhancement New feature or request model Related to the classifier model

Comments

@ikmckenz
Copy link
Owner

ikmckenz commented Aug 25, 2020

On the data set with ChEMBL 25, with the current (53ec926) data cleaning/grouping. With 10 trees on the sklearn random forest we get 77% on top-1 precision, recall, and F1 score, and 89-92% top-5 accuracy.

@ikmckenz ikmckenz added enhancement New feature or request model Related to the classifier model labels Aug 25, 2020
@ikmckenz
Copy link
Owner Author

In 833ece7 I added a Jupyter notebook that implements a very basic neural network classifier model. It gets 79% on top-1 accuracy, and 97% top-5 accuracy.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request model Related to the classifier model
Projects
None yet
Development

No branches or pull requests

1 participant