Slide 3
Slide 3 text
2
Feature importanceの実装
• Feature importance の2種類の定義
• 分割に使われた回数(split)
• 分割において得られた情報ゲインの総和(gain) ← こちらを採⽤
class Node:
…
def feature_importance(self, importance_value):
if not self.is_leaf:
importance_value[self.split_feature_index] += self.gain
self.left_node.feature_importance(importance_value)
self.right_node.feature_importance(importance_value)
return importance_value
ルートノードから
情報ゲインを再帰的に
⾜し合わせていく