ランキングにxgboostを使用しています
param = {'objective':'rank:pairwise', 'booster':'gbtree'}
私が理解しているように、勾配ブースティングは、学習した決定木の重み付き合計を計算することによって機能します。学習した各ブースターに割り当てられた重みにアクセスするにはどうすればよいですか? トレーニング後に重みを後処理して予測ステップを高速化したかったのですが、個々の重みを取得する方法がわかりません。を使用するdump_model()
と、作成されたファイルにさまざまなディシジョン ツリーが表示されますが、重み付けは保存されません。API で、適切な関数が見つかりませんでした。または、収縮パラメータを使用して手動で重みを計算できますeta
か?