-
Notifications
You must be signed in to change notification settings - Fork 28.4k
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
[SPARK-11401] [MLLIB] PMML export for Logistic Regression Multiclass Classification #9397
[SPARK-11401] [MLLIB] PMML export for Logistic Regression Multiclass Classification #9397
Conversation
Jenkins, ok to test |
It's hard to see the changes with moving the file from |
Test build #44801 has finished for PR 9397 at commit
|
Here the diff between the two files (BinaryClass vs Class): Practically the difference between the Binary and the Class version is that instead of building two fixed regression tables (YES/NO), I create a Category Zero table (the one without the predictors) and as many regression tables as the number of the categories (numClasses-1). The code, in case of numClasses = 2, does exactly what it was doing before in the BinaryClassificationPMMLModelExport class that I originally wrote. My spark validator project confirms both Binary and Multiclass pmml export works fine: See sections: |
okay. please fix fails MiMa tests. thanks. |
@dbtsai
Not familiar with Mima. Thanks. |
You need to add |
Test build #45152 has finished for PR 9397 at commit
|
Test build #45306 has finished for PR 9397 at commit
|
@dbtsai I don't think the issue with failed Spark test is to do with my code. |
Test build #45877 has finished for PR 9397 at commit
|
Jenkins, retest this please |
Test build #46554 has finished for PR 9397 at commit
|
@dbtsai any advice on why it is failing? All the pmml tests passed and that is the only thing I changed. |
Can you rebase the master? |
generalized classification pmml model export to support both multiclass and binary classification
96a19d6
to
fd38551
Compare
Test build #46784 has finished for PR 9397 at commit
|
@dbtsai thanks for the suggestion, rebasing from master seems to have fixed it. |
Thanks for the pull request. I'm going through a list of pull requests to cut them down since the sheer number is breaking some of the tooling we have. Due to lack of activity on this pull request, I'm going to push a commit to close it. Feel free to reopen it or create a new one. @dbtsai there are a few pull requests that were waiting on your review. Can you revisit them even if they are closed? |
Jira issue: https://issues.apache.org/jira/browse/SPARK-11401