-
Notifications
You must be signed in to change notification settings - Fork 28.1k
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-28045][ML][PYTHON] add missing RankingEvaluator #24869
Conversation
Test build #106497 has finished for PR 24869 at commit
|
Test build #106498 has finished for PR 24869 at commit
|
Test build #106500 has finished for PR 24869 at commit
|
Test build #106503 has finished for PR 24869 at commit
|
Test build #106505 has finished for PR 24869 at commit
|
friendly ping @srowen |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks generally OK to me
setDefault(metricName -> "meanAveragePrecision") | ||
|
||
final val k = new IntParam(this, "k", | ||
"The ranking position value used in meanAveragePrecisionAtK|precisionAtK|ndcgAtK|recallAtK." + |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not a big deal, but could you generate this from the supportedMetrics
by filtering those including "AtK"?
9f505c0
to
f9b6777
Compare
Test build #106698 has finished for PR 24869 at commit
|
import org.apache.spark.mllib.util.MLlibTestSparkContext | ||
import org.apache.spark.mllib.util.TestingUtils._ | ||
|
||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There are many extra blank lines below, but I wouldn't hold up merging it for that.
Test build #106873 has finished for PR 24869 at commit
|
Merged to master |
## What changes were proposed in this pull request? add missing RankingEvaluator ## How was this patch tested? added testsuites Closes apache#24869 from zhengruifeng/ranking_eval. Authored-by: zhengruifeng <ruifengz@foxmail.com> Signed-off-by: Sean Owen <sean.owen@databricks.com>
this is really great to see @zhengruifeng @srowen , can't wait to try this out |
This looks great! Sorry, maybe I am overlooking something but is this supposed to be used with the .transform() or .recommendForAllUsers() API in the case of ALS? |
What changes were proposed in this pull request?
add missing RankingEvaluator
How was this patch tested?
added testsuites