Gradient Boosted Decision Tree の python 実装です。
アルゴリズムのコアな部分は numpy のみを用いて実装されています。
- Introduction to Boosted Trees
- Gradient Boosted Tree (Xgboost) の取り扱い説明書
- Gradient Boosting のアルゴリズムの詳細
- 必要なもの (Requirement)
- 使い方 (Usage)
- 実際の例 (Example)
- MNISTの分類（binary_classification)
- 人工データによる分類（二値分類、回帰問題）
サンプルの実行には以下のライブラリが必要です
numpy
scikit-learn
matplotlib
pandas
scipy
事前にホストマシン上に docker 及び docker-compose がインストールされていることが条件です。
まず docker-compose を用いてイメージを build, その後コンテナを daemon で起動しておきます。
docker-compose build
docker-compose up -dサンプルのコマンドはコンテナ内部で実行します
# コンテナの内部に潜り込む
docker exec -it gbdt-app bash
# sample.py を実行
python sample.pyvenv を使うのがいいかなと思います。
python3 -m venv .venv
source ./.venv/bin/activate
pip install -U pip && pip install -r requirements.txtgit clone もしくはdownloadしたフォルダを実行ファイルと同じ階層に置きます
import gbdtree as gb
clf = gb.GradientBoostedDT()
x_train,t_train = ~~ # 適当なトレーニングデータ
clf.fit(x=x_train, t=t_train)mnist.py と sample.py の2つのファイルが実行サンプルになっています。
MNIST の手書きデータを用いた分類問題をときます。
-
training data
MNIST Originalの手書き文字データ * 出力は{0, 1, 2,..., 9}の１０クラス分類問題- そのままだと時間がかかりすぎるので、二値分類（３と８の分類）で
datasize=5000になおして実行
-
Gradient Boosted Tree の parameters
- 目的関数：交差エントロピー
- 活性化関数：ロジスティクスシグモイド関数
Note:
mnist.pyではMNISTの手書きデータ・セットをネット上から取得するので、ローカルにデータを持っていない場合にかなり時間がかかる場合があります。また学習時間もパラメータをデフォルトのままで行うと30分ぐらいかかります。計算を投げてご飯でも食べに行きましょう。
- feature_importance
- 学習時の logging
が /examples/mnist に出力されます.
start build new Tree
build new node depth=0_N=2000 gain=538.7344
build new node depth=1_N=1000 gain=96.3745
build new node depth=1_N=1000 gain=45.4259
build new node depth=2_N=163 gain=40.5793
build new node depth=2_N=855 gain=22.9140
build new node depth=2_N=145 gain=21.3546
build new node depth=2_N=837 gain=19.1825
build new node depth=3_N=117 gain=19.0716
build new node depth=3_N=82 gain=15.8894
build new node depth=3_N=824 gain=12.5804
build new node depth=3_N=26 gain=9.2701
build new node depth=4_N=815 gain=7.2187
build new node depth=4_N=28 gain=6.4043
build new node depth=3_N=81 gain=5.9525
build new node depth=4_N=12 gain=5.7942
==============================
end tree iteration
iterate:0 loss:4.14e-01
valid loss: 4.308e-01
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
build new node depth=0_N=5000 gain=0.0131
build new node depth=1_N=2593 gain=0.0306
build new node depth=2_N=2392 gain=0.0351
build new node depth=3_N=1911 gain=0.0245
build new node depth=3_N=481 gain=0.0229
build new node depth=4_N=301 gain=0.0197
build new node depth=4_N=180 gain=0.0196
build new node depth=1_N=2407 gain=0.0170
build new node depth=2_N=772 gain=0.0245
build new node depth=3_N=366 gain=0.0389
build new node depth=2_N=1635 gain=0.0245
build new node depth=3_N=1348 gain=0.0154
build new node depth=4_N=202 gain=0.0148
build new node depth=3_N=287 gain=0.0145
build new node depth=3_N=406 gain=0.0145
reach to 15 nodes. stop build node.
==============================
end tree iteration
iterate:99 loss:2.96e-04
(improve: 2.401e-05)
valid loss: 2.941e-02
validation accuracy: 0.9898505465090341
最終的な validation accuracy は 98.98%
feature importance (gain) を対数にとって画像にしたもの. 3 と 8 でずれる画像左の真ん中が多めに選ばれてそう?
実行方法は単に python スクリプトとして実行すればOKです。引数等はありません。
python sample.py実行すると以下の2つの問題を解きます
- 人工的に作成した二次元入力に対する二値分類問題
- 人工的に作成した一次元入力に対する実数値の回帰問題
- training data:
- 各クラスを、[1,1] [-1.,-1] を中心としたガウス分布からのサンプリングから作成します * 図中で青と緑で表示されています.
- モデルパラメータ
- 目的関数: 交差エントロピー
- 活性化関数.: シグモイド関数
一次元のランダムな入力に対して、正解関数 + ノイズを付与した正解ラベルを作成し、それを予測するようなモデルを作成します。この時複数の boosting 回数でモデルを作成し、回数が多くなると予測値がよりデータに引っ張られていく様子を可視化します.
boosting の回数は n_iter で制御されている為これを変化させて学習機をそれぞれの n_iter で作成し, 予測値をグラフにプロットしています.
- training data
- 以下で定義される関数値にガウスノイズを加えたもの
def test_function(x):
return 1 / (1. + np.exp(-4 * x)) + .5 * np.sin(4 * x)- モデルパラメータ
- 目的関数： 二乗ロス関数
- 活性化関数: 恒等写像
max_depth=8（毎回最大でどのぐらいの深さまで木を作るか）gamma=.01（木が成長できる最小のgainを規定するパラメータ)lam=.1
結果
gamma や max_depth を変えたり, valid_data を作って, train/valid loss を可視化してみるのも面白いかも知れません。