如何将图像数据集导入autokeras程序

2024-04-20 15:25:33 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用自动keras来创建一个模型,可以在两种不同皮肤疾病的图片之间进行识别。我对如何将数据集导入代码有问题。你知道吗

我的代码基于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示例代码时类似的结果。你知道吗


Tags: 数据代码fromtestimporttrue示例train