загрузка модели автокера с временно сохраненными двоичными файлами

Question

Как видно из этого поста , существует проблема с сохранением и загрузкой моделей автокера (для получения более подробной информации обратитесь к предоставленной ссылке).Я заметил, что во временной папке autokeras есть несколько двоичных файлов (например, / tmp / autokeras_ECWBT7), которые содержат несколько двоичных файлов:

graph (1.graph, 2.graph, ...)

best_model.txt

классификатор

module

temp_model

Я не смог найти никакой документации, чтобы увидеть, как я могу использовать это для загрузки обученной модели (еслиесть какие-либо).

Буду признателен, если вы поможете мне в этой проблеме.

Answer 1

Autokeras действительно не хватает документации.Я нашел несколько способов, как обученную модель можно использовать в дальнейшем (для autokeras версии 0.3.7) в случае ImageClassifier:

1. , если вы используете autokeras.image.image_supervised.ImageClassifier и хотите продолжить поиск моделей, используяfit, или перезапустите final_fit для лучшей модели в вашем Graph_Directory , вы можете использовать это:

clf = ImageClassifier(verbose=True, augment=False, path='./path/to/graph/directory', resume=True)

теперь вы можете продолжить поиск в сетевой архитектуре или экспортировать свою лучшую модель.Если вы хотите экспортировать свою лучшую модель ImageClassifier и использовать ее для прогнозирования, вы можете использовать: clf.export_autokeras_model('your_model.pkl')

и позже вы можете загрузить ее таким образом:

from autokeras.utils import pickle_from_file

model = pickle_from_file('your_model.pkl')

и после этого вы можете делать прогнозы по: y_predict = model.predict(x_test)

Это подходящее решение,если вы хотите позже работать именно с моделью Autokeras ImageClassifier (не конвертировать ее во что-то еще).