如何从文件中导入(恢复)用tflearn构建的神经网络模型
自定义文本分类模型的保存与生成文件记录
我最近跟着一篇文本分类教程动手实践,自己构建了一个自定义文本分类训练集,之后用以下代码完成模型的定义、训练和保存:
# Define model and setup tensorboard model = tflearn.DNN(net, tensorboard_dir='tflearn_logs') # Start training (apply gradient descent algorithm) model.fit(train_x, train_y, n_epoch=1000, batch_size=8, show_metric=True) model.save('model.tflearn')
运行这段代码后,会生成以下几个相关文件:
model.tflearn.data-00000-of-00001model.tflearn.indexmodel.tflearn.meta
内容的提问来源于stack exchange,提问作者Morse
相关产品推荐
相关产品推荐

