例如如何保存此模型:https://github.com/tensorflow/tensorflow/blob/r1.2/tensorflow/examples/tutorials/input_fn/boston.py
答案 0 :(得分:1)
当您致电时,模型会保存在model_dir
中:
regressor = tf.contrib.learn.DNNRegressor(feature_columns=feature_cols,
hidden_units=[10, 10],
model_dir="/tmp/boston_model")
现在在推理过程中,您使用新输入调用上述内容然后调用regressor.predict()
函数。