首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >TFLearn不加载模型

TFLearn不加载模型
EN

Data Science用户
提问于 2016-12-20 15:02:36
回答 1查看 3.2K关注 0票数 0

我试着学习并掌握学习的诀窍。我尝试过的一件事是保存我的模型,然后重新加载它。我收到一个错误,上面说:

代码语言:javascript
复制
W tensorflow/core/framework/op_kernel.cc:975] Not found: Unsuccessful TensorSliceReader constructor: Failed to find any matching files for quicktest.model

下面是我的模型以及通过保存模型而创建的检查点文件。我搜索了这个错误,并尝试了许多不同的文件更改,但是没有什么能修复它。

代码语言:javascript
复制
    import tflearn
    data, labels = load_csv('my_csv', target_column=0, categorical_labels=True, n_classes=2)
    net = tflearn.input_data(shape=[None, 32])
    net = tflearn.fully_connected(net, 32)
    net = tflearn.fully_connected(net, 2, activation='softmax')
    net = tflearn.regression(net)
    model = tflearn.DNN(net)
    model.fit(data, labels, n_epoch=1, batch_size=10, show_metric=True)
    model.save('quicktest.model')
    model.load('quicktest.model')

检查点

代码语言:javascript
复制
model_checkpoint_path: "quicktest.model"
all_model_checkpoint_paths: "quicktest.model"
EN

回答 1

Data Science用户

发布于 2016-12-20 15:10:44

找出问题所在:

代码语言:javascript
复制
model.load('filename')

实际上应该是:

代码语言:javascript
复制
model.load('./filename')

希望这能帮到别人。

票数 1
EN
页面原文内容由Data Science提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://datascience.stackexchange.com/questions/15833

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档