Как я могу использовать сохраненные веса, чтобы делать прогноз в следующий раз без какого-либо процесса обучения?

Question

Я обучил CNN для обнаружения пневмонии.Я сохранил веса обученной модели на том же пути.Теперь я не могу использовать сохраненную модель для прогнозирования данных после перезапуска программы.

Я попытался выполнить определенные строки кода, которые отвечают за прогнозирование, но затем я получаю ошибку импорта, из которой я могу понять, что мне нужно перезапустить всю программу, которая снова требует времени.Это фактический код предсказания.

res = classifier.predict(test_image)
if res[0][0] == 1:
        prediction = 'pnemonia'
elif res[0][0] == 0:
        prediction = 'normal'
print(prediction)

Фактическим результатом должно быть предсказание.Вместо этого я получаю

Using TensorFlow backend.
Traceback (most recent call last):
  File "<input>", line 5, in <module>
NameError: name 'np' is not defined

Answer 1

np обычно является сокращением для модуля Numpy.Держу пари, что вам нужно import numpy as np в модуле, где вы получаете сообщение об ошибке.Возможно, импорт не вызывается, потому что вы больше не используете другой модуль для генерации весов.