Блог

Главная - Вопросы по программированию - Python TFLearn не удается сохранить в один файл model.tfl

Python TFLearn не удается сохранить в один файл model.tfl

#python #python-3.x #deep-learning #model #tflearn

#питон #python-3.x #глубокое обучение #Модель #tflearn

Вопрос:

Вот мой код:

 try:
     model.load("./model.tfl")
except:
     tensorflow.compat.v1.reset_default_graph()

     net = tflearn.input_data(shape=[None, len(training[0])])
     net = tflearn.fully_connected(net, 8)
     net = tflearn.fully_connected(net, 8)
     net = tflearn.fully_connected(net, 8)
     net = tflearn.fully_connected(net, len(output[0]), activation="softmax")
     net = tflearn.regression(net)

     model = tflearn.DNN(net)

     model.fit(training, output, n_epoch=1000, batch_size=8, show_metric=True)
     model.save("model.tfl")

Моя проблема в том, что всякий раз, когда я сохраняю модель, это приводит к созданию 3 файлов:

  • модель.tfl.данные-00000-of-00001
  • модель.tfl.index
  • модель.tfl.meta

Как бы мне сохранить их в один файл? Или как мне загрузить модель из этих файлов?

Комментарии:

1. Попробуйте сохранить модель, используя расширение ‘.model’. то есть: model.save("model_name.model"); для загрузки модели: model.load("model_name.model")

2. @MDBillalHossain Я уже пробовал это, независимо от того, какое я делаю расширение или имя файла, он все равно делает то же самое