如何能够保存没有变量的张量流模型?

时间:2017-06-27 13:37:41

标签: machine-learning tensorflow

1 个答案:

答案 0 :(得分:1)

当您致电时,模型会保存在model_dir中:

regressor = tf.contrib.learn.DNNRegressor(feature_columns=feature_cols,
                                        hidden_units=[10, 10],
                                        model_dir="/tmp/boston_model")

现在在推理过程中,您使用新输入调用上述内容然后调用regressor.predict()函数。