我正在尝试使用自动角膜模型创建一个模型,该模型可以识别两种不同皮肤疾病的照片。我在将数据集导入代码中遇到问题。
我的代码基于mnist数据集的auto keras文档中的示例代码。
from keras.datasets import mnist
from autokeras.classifier import ImageClassifier
if __name__ == '__main__':
(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, augment=False)
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 * 100)
与运行mnist示例代码时一样,我希望输出在终端上给我类似的结果。