tensorflow tensorflow.contrib.learn.Estimator加载训练模型

Question

我这样调用这个函数后：

    from tensorflow.contrib import learn
    #----------------------------------------
    #Do some process here
    #----------------------------------------
    classifier = learn.Estimator(model_fn=bag_of_words_model,model_dir='F:/data')
    classifier.fit(feature_train, target_train, steps=1000)

我的文件夹“F：/ data”中会有这样的文件

我想知道我还有重用这个模型吗？就像转移到新计算机并使用它来预测新数据一样。对不起，我的英语不好。感谢所有的答案！！希望你们都过得愉快。

Answer 1

在要重复使用模型的脚本中，再次重新定义/导入bag_of_words_model并定义

classifier_loaded = learn.Estimator(model_fn=bag_of_words_model, model_dir='F:/data')

Tensorflow将重新加载图表并将权重导入图表中，您可以将其用作

classifier_loaded.predict(input_fn=input_fn)

或继续培训

claissifier_loaded.fit(feature_train, target_train)