code-intelligence icon indicating copy to clipboard operation
code-intelligence copied to clipboard

Store and use class labels in MLPClassifier

Open jlewi opened this issue 5 years ago • 1 comments

It looks like our repo_specific_model is using MLPClassifier. https://scikit-learn.org/stable/modules/generated/sklearn.neural_network.MLPClassifier.html

The labels, however, are not stored in MLPClassifier's classes attribute. Instead the label's are stored in the YAML file describing the model.

The code assumes that the labels in the YAML file are listed in the same order as the probabilities generated by the MLPClassifier for predictions.

It would probably be better to store the labels in the MLPClassifier to decrease the risk of them getting out of sync.

jlewi avatar Dec 30 '19 15:12 jlewi

Issue-Label Bot is automatically applying the label kind/feature to this issue, with a confidence of 0.79. Please mark this comment with :thumbsup: or :thumbsdown: to give our bot feedback!

Links: app homepage, dashboard and code for this bot.

issue-label-bot[bot] avatar Dec 30 '19 15:12 issue-label-bot[bot]