Google Colaboratory で Auto Kerasを実行する

昨日投稿して初めていいね貰ったので気を良くしてGoogle Colaboratoryでの実験結果など書いていきましょう。

最近は機械学習プロセスである

-ハイパーパラメーターチューニング

-モデル選択

-特徴エンジニアリング

を対象としてプロセスを自動化してくれるソフトが出ており

そのうちのOSSであるAuto KerasをGoogle Colaboratoryで使用出来たら

便利だろうと思いましたので試してみました

Auto KerasはPytorchを使用する為Pytorchをインストールします。

太文字はColaboratory上のJupyter Notebookのコードセルに打ち込むコマンドです。

!pip3 install torch torchvision

Requirement already satisfied: torch in /usr/local/lib/python3.6/dist-packages (1.0.1)

Requirement already satisfied: torchvision in /usr/local/lib/python3.6/dist-packages (0.2.1)

Requirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from torchvision) (1.15.4)

Requirement already satisfied: pillow>=4.1.1 in /usr/local/lib/python3.6/dist-packages (from torchvision) (4.1.1)

Requirement already satisfied: six in /usr/local/lib/python3.6/dist-packages (from torchvision) (1.11.0)

Requirement already satisfied: olefile in /usr/local/lib/python3.6/dist-packages (from pillow>=4.1.1->torchvision) (0.46)

インポートがエラーなくされるか確認、特に何も出力されなければ成功

import torch

Auto Kerasをインストールします。

!pip install autokeras

その後試しに下記のAuto KerasでMNISTを学習させるサンプルコードを実行して見ました。

from keras.datasets import mnist
import autokeras
from autokeras.image.image_supervised import ImageClassifier
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train = x_train.reshape(x_train.shape + (1,))
x_test = x_test.reshape(x_test.shape + (1,))
clf = ImageClassifier(verbose=True)
clf.fit(x_train, y_train, time_limit=12 * 60 * 60)
clf.final_fit(x_train, y_train, x_test, y_test, retrain=True)
y = clf.evaluate(x_test, y_test)
print(y)

⬇️実行すると下記の様にモデルの学習と保存が繰り返し行われます。

Using TensorFlow backend.

Saving Directory: /tmp/autokeras_W2FY89

Preprocessing the images.

Preprocessing finished.

Initializing search.

Initialization finished.

+----------------------------------------------+

| Training model 0 |

+----------------------------------------------+

No loss decrease after 5 epochs.

Saving model.

+--------------------------------------------------------------------------+

| Model ID | Loss | Metric Value |

+--------------------------------------------------------------------------+

| 0 | 0.3519867166876793 | 0.9672000000000001 |

+--------------------------------------------------------------------------+

+----------------------------------------------+

| Training model 1 |

+----------------------------------------------+

No loss decrease after 5 epochs.

Saving model.

+--------------------------------------------------------------------------+

| Model ID | Loss | Metric Value |

+--------------------------------------------------------------------------+

| 1 | 0.1383698031306267 | 0.9868 |

+--------------------------------------------------------------------------+

+----------------------------------------------+

| Training model 2 |

+----------------------------------------------+

Epoch-15, Current Metric - 0.992: 99%|██████████████████████▊| 460/465 [08:46<00:05, 1.20s/ batch]

一応稼働は確認しましたがめっちゃ時間かかってます。２時間くらいは回しましたがまだ終わりません。

参考