利用官网中的代码进行测试:
from keras.datasets import mnist
from autokeras 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)
ps:会先输入这样的结果
等运行到:clf.fit(x_train, y_train, time_limit=12 * 60 * 60)
会开始搜索相对最好的网络结构(一个一个model的训练),