Skip to content
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

Possible to support LGBMRanker? #3

Closed
travisbrady opened this issue Apr 11, 2017 · 4 comments
Closed

Possible to support LGBMRanker? #3

travisbrady opened this issue Apr 11, 2017 · 4 comments

Comments

@travisbrady
Copy link

Is it possible to support https://github.com/Microsoft/LightGBM/blob/master/python-package/lightgbm/sklearn.py#L639 ?
If so I'd be happy to contribute, just not sure if it's even doable.

thanks

@vruusmann
Copy link
Member

The doability depends on the definition of the "lambdarank" objective function - this is a piece of code that transforms the raw GBT score to the final user score.

Once known, create an appropriate subclass of org.jpmml.lightgbm.ObjectiveFunction and register it with the org.jpmml.lightgbm.GBDT#parseObjectiveFunction(...) method:
https://github.com/jpmml/jpmml-lightgbm/blob/master/src/main/java/org/jpmml/lightgbm/GBDT.java#L282

@travisbrady
Copy link
Author

Got it, thank you. I'll do some investigating.

@mohit-shrma
Copy link

Do you have any update on this?

Exception in thread "main" java.lang.IllegalArgumentException: lambdarank                                
        at org.jpmml.lightgbm.GBDT.parseObjectiveFunction(GBDT.java:338)                                 
        at org.jpmml.lightgbm.GBDT.load(GBDT.java:82)                                                    
        at org.jpmml.lightgbm.LightGBMUtil.loadGBDT(LightGBMUtil.java:58)                                
        at org.jpmml.lightgbm.LightGBMUtil.loadGBDT(LightGBMUtil.java:50)                                
        at org.jpmml.lightgbm.Main.run(Main.java:122)                                                    
        at org.jpmml.lightgbm.Main.main(Main.java:115)  

@ancasarb
Copy link

Are there any updates on when/if this will be supported? Thank you in advance!

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

No branches or pull requests

4 participants