AutoML③(train_size変更)

train_size変更

訓練データの分割の割合は、setup関数の引数train_sizeで変更可能です。(6行目)

(またasilentにTrueを指定することで型確認のダイアログをスキップすることができます。)(7行目)

[Google Colaboratory]

1
2
3
4
5
6
7
ret = setup(boston_data,
target = "medv",
session_id=0,
normalize = False,
numeric_features = ["chas"],
train_size = 0.8,
asilent=True)

今回はtrain_size0.8を指定したので、訓練データが80%、テストデータが20%になります。

[実行結果]

Transformed Train SetTransformed Test Setのサイズが変わっていることが確認できます。