search

dmlc / xgboost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow
https://xgboost.readthedocs.io/en/stable/
Apache License 2.0
26.3k stars 8.73k forks source link

Is there an implementation of LambdaMART model? #30

Closed RalphAdu closed 10 years ago

tqchen commented 10 years ago

Yes, the rank module is designed to be lambdaRank, see the incoming new branch https://github.com/tqchen/xgboost/tree/unity https://github.com/tqchen/xgboost/blob/unity/src/learner/objective.h

However, in our test on the ranking data-set, we didn't manage to make lambdaRank output-perform pairwise ones, this makes us hold the release of current code.

tqchen commented 10 years ago

The unity branch is now in master

RalphAdu commented 10 years ago

thanks for share the code