PreTrainingChainをscikit-learnのGridSearchに対応しました
以前作ったChainerのラッパーであるPreTrainingChain
chainerで気軽にスケールできる事前学習器付きニューラルネット生成 - verilog書く人
をscikit-learnの学習器ライクに使えるようにしたのとGridSearchに対応しました。
ChainClassfierをインスタンシエートして、
ChainClassfier.fit(x_train, y_train)で学習、ChainClassfier.predict(x_test)で予測、
ChainClassfier.score(x_test, y_test)でテストデータに対するスコアを出すようにしました。
以下がGridSearchによるパラメータチューニングの例です。
pip install PreTrainingChainでインストールできます。お試しあれ。
わたしのパソコンは恐ろしく熱くなりました。話題のディープヒーティングです。
追記:
chainer 1.8.0に更新したら動かなくなったのでアップデートしました。