LightGbm和catboost的Elasticsearch LTR插件示例模型和功能?

quhf5bfb  于 2023-10-17  发布在  ElasticSearch
关注(0)|答案(1)|浏览(96)

我正在尝试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格式。

vltsax25

vltsax251#

根据this issuethis one判断,我认为LightGBM或CatBoost都不支持。
然而,在某些情况下(只有数字特征的密集向量),似乎有一种方法可以使用lgb2xgb将LightGBM转换为XGBoost。

相关问题