Skip to content

[ML] Extend classification to support multiple classes #53539

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

Conversation

dimitris-athanasiou
Copy link
Contributor

Prepares classification analysis to support more than just
two classes. It introduces a new parameter to the process config
which dictates the num_classes to the process. It also
changes the max classes limit to 30 provisionally.

@elasticmachine
Copy link
Collaborator

Pinging @elastic/ml-core (:ml)

@dimitris-athanasiou dimitris-athanasiou force-pushed the extend-classification-to-support-multiclass branch from 10c4d7e to 367cfd6 Compare March 13, 2020 13:01
Prepares classification analysis to support more than just
two classes. It introduces a new parameter to the process config
which dictates the `num_classes` to the process. It also
changes the max classes limit to `30` provisionally.
@przemekwitek przemekwitek self-requested a review March 16, 2020 08:20
@@ -272,7 +275,7 @@ public boolean supportsCategoricalFields() {
@Override
public List<FieldCardinalityConstraint> getFieldCardinalityConstraints() {
// This restriction is due to the fact that currently the C++ backend only supports binomial classification.
return Collections.singletonList(FieldCardinalityConstraint.between(dependentVariable, 2, 2));
return Collections.singletonList(FieldCardinalityConstraint.between(dependentVariable, 2, 30));
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should 30 become a constant in this class?

Copy link
Contributor

@przemekwitek przemekwitek left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@dimitris-athanasiou dimitris-athanasiou merged commit 7e0a3c4 into elastic:master Mar 16, 2020
@dimitris-athanasiou dimitris-athanasiou deleted the extend-classification-to-support-multiclass branch March 16, 2020 11:51
dimitris-athanasiou added a commit that referenced this pull request Mar 16, 2020
…53597)

Prepares classification analysis to support more than just
two classes. It introduces a new parameter to the process config
which dictates the `num_classes` to the process. It also
changes the max classes limit to `30` provisionally.

Backport of #53539
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants