Upgrade to Pro — share decks privately, control downloads, hide ads and more …

特徴量の重要度計算を実装する ランダムフォレストを実装する

特徴量の重要度計算を実装する ランダムフォレストを実装する

Transcript

  1. 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 ルートノードから 情報ゲインを再帰的に ⾜し合わせていく