dmlc / dmlc/xgboost

Missing XGBoostRanker in xgboost4j-spark jvm package

Open
#10,118 3 comments 0 reactions 0 assignees View on GitHub
feature-request
Dominant language
C++
Stars
28.8k
Forks
8.9k
Avg merge
1d 12h
Merged PRs (30d)
54

Description

On the latest version (2.03) of xgboost4j-spark XGBoostClassifer and XGBoostRegressor are available for classification and regression trainning tasks, but there seems to be no equivalent pipeline for learning-to-rank tasks. How should I train a ranking model using current pipelines or lower level apis?Has anyone successfully run it on spark before?

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue names the xgboost4j-spark JVM package, XGBoostClassifier, XGBoostRegressor, current pipelines, and lower-level APIs; start by reading those existing Spark entry points and checking how ranking data is represented. Determine whether a learning-to-rank path can use the current APIs or requires a new XGBoostRanker, and validate the result with a working Spark ranking-training example.

Written by the indexing model from the issue text.

Assessment

Tech stack
java, scala, spark
Domain
distributed-systems, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.