機械学習理論研究室 ← AI体験の一覧

質問で分ける木

「気温は28℃より高い?」——こんな「はい/いいえ」の質問を重ねるだけで、コンピュータはものごとを分類できます。この仕組みを決定木といいます。決定木が自分で質問を選び、データの地図を切り分けていく様子を見てみましょう。

木が1段ずつ育つ

ボタンを押すたびに、質問が1つ増えて、地図が1本の線で切られます

気温湿度だけをメモした、ある年の記録があります。その日を「暑い」と感じたかどうかを、この2つの数字だけで当ててみます。点1つが1日ぶんの記録です。

暑い 暑くない
決定木(質問の並び)
同じデータを地図にすると

木の四角や丸をタップすると、それが担当している地図の範囲が光ります。木の箱ひとつが、地図の範囲ひとつにぴったり対応しています。

まだ1回も分割していません。すべての日がひとつのグループに入っていて、暑い日と暑くない日が混ざっています。
分割した回数 0
葉(答えの箱)の数 1
全体のジニ係数 0.000
正しく当てられた割合 0%

混ざり具合を1つの数で表す

g = 1 − Σ(各クラスの割合)²

青3個・赤2個のグループなら、割合は 0.6 と 0.4。式に入れると g = 1 −(0.6² + 0.4²) = 0.48 です。全部同じ色なら g = 0。この g をジニ係数といい、そのグループがどれだけ混ざっているかを表します。

0.5に近い=ぐちゃぐちゃ/0に近い=すっきり純粋。 決定木は、分割したあとのジニ係数(左右の大きさで重みを付けた平均)がいちばん小さくなる質問を選びます。

選ばれた質問分割前 g分割後 g(加重平均)下がった分