我正在尝试elasticsearch ltr插件。我在插件文档中发现了一个线性模型,它的JSON格式如下所示:
POST _ltr/_featureset/performancetest/_createmodel
{
"model": {
"name": "lightgbm_model",
"model": {
"type": "model/gbdt",
"definition": """
{
"num_leaves": 31,
"objective": "lambdarank",
"boosting_type": "gbdt",
"metrics": "ndcg",
"ndcg_eval_at": [5, 10, 20]
}
"""
}
}
}
但是我找不到类似的lightgbm和catboost的例子。任何人都可以分享一些小的示例模型和功能文件,因为据我所知,插件接受的格式是每个模型的特定JSON格式。
1条答案
按热度按时间vltsax251#
根据this issue和this one判断,我认为LightGBM或CatBoost都不支持。
然而,在某些情况下(只有数字特征的密集向量),似乎有一种方法可以使用lgb2xgb将LightGBM转换为XGBoost。