在Tensorflow中保存模型

时间:2018-07-05 10:48:37

标签: python-3.x tensorflow

model = tf.estimator.LinearClassifier(feature_columns=feat_cols)
model.train(input_fn=input_func, steps=10000)

我使用LinearClassifier模型训练了我的数据。现在,每次使用新数据进行测试时,都会花费太多时间,因此,我不想再次进行培训。所以我想保存模型并要使用不止一次。我怎样才能做到这一点? 我看到了解决方法

saver = tf.train.Saver()

但是我该如何在我的模型中实现

0 个答案:

没有答案