木の分割
変数AまたはBで分割した場合に得られるGini指数が与えられています。次にどの変数で木を分割すべきかを判断してください。
この演習はコースの一部です
HRアナリティクス:Pythonで従業員離職を予測する
演習の手順
- どちらのGiniが小さいかを確認します。
- AのGiniが小さければAで分割し、そうでなければBで分割します。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Gini index in case of splitting by variable A or B
gini_A = 0.65
gini_B = 0.15
# check which Gini is lower and use it for spliting
if ____ < ____:
print("split by A!")
____:
print("split by B!")