Chapter 06

Random Forestを用いた訓練とモデルの保存

(株)ディープブラック
(株)ディープブラック
2021.03.21に更新
このチャプターの目次

Random Forestを用いた訓練

scikit-learnのRandom Forestを用いるには、以下のように書きます。

######### RandomForest #########
clf = RandomForestClassifier(max_depth=2, random_state=0)
clf.fit(data.values, target)

Modelの保存

pickleでRandom Forestとcategory_encodersを保存します。

######### Modelの保存 #########
with open('models/clf_ohe.pkl', 'wb') as f:
	pickle.dump([clf, ce_ohe], f)