始める無料で始める

ジニ指数を計算する

決定木アルゴリズムは、終端ノードの純度ができるだけ高くなるように分割を行います。ジニ指数はそのために用いられる指標の一つで、各グループのサンプル比率に基づいて計算します。

それぞれのグループに属する、残った人(stayed)と離職した人(left)の人数が与えられているとき、そのノードのジニ指数を計算してください。

この演習はコースの一部です

HRアナリティクス:Pythonで従業員離職を予測する

コースを見る

演習の手順

  • そのノードに含まれる従業員の合計人数を計算します。
  • 各グループの従業員比率に基づいてジニ指数を計算します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

#number of people who stayed/left
stayed = 37
left = 1138

#sum of stayed and left
total = ____ + ____

#gini index
gini = ____*(____/total)*(____/total)
コードを編集して実行