SSブログ

[R]LightGBMを使った特徴量重要度の出し方 [データサイエンス、統計モデル]

pythonの資料は豊富に見つかるのですが、意外とRのLightGBMを使った方法が見つからなかったのでメモとして記載しておきます。
作成したモデル(モデル名:model)に対して、
lgb.importanceで重要度を計算してくれます
lgb.plot.importanceで重要度の可視化をしてくれます。

#モデリング
model <- lightgbm(data = dtrain,
nrounds = *****,
min_data_in_leaf = *****,
learning_rate = *****,
params = ***** );

#重要度の作成
tree_imp <- lgb.importance(model, percentage = TRUE)

#重要度の可視化
lgb.plot.importance(tree_imp, top_n = 5L, measure = "Gain")

nice!(3)  コメント(0) 
共通テーマ:学問

nice! 3

コメント 0