Как импортировать набор данных изображений в программу auto-keras - PullRequest
Новая
       36

Как импортировать набор данных изображений в программу auto-keras

2 голосов
/ 06 апреля 2019

Я пытаюсь использовать авто-керас для создания модели, которая может определять между изображениями двух разных кожных заболеваний.У меня проблема с тем, как импортировать мой набор данных в код.

Я основал свой код на примере кода из документации auto keras для набора данных mnist.

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.

...