例如,如何保存这个模型: https://github.com/tensorflow/tensorflow/blob/r1.2/tensorflow/examples/tutorials/input_fn/boston.py
回答:
模型在调用时被保存到 model_dir
中:
regressor = tf.contrib.learn.DNNRegressor(feature_columns=feature_cols, hidden_units=[10, 10], model_dir="/tmp/boston_model")
在推理过程中,你可以调用上述代码,然后使用新的输入调用 regressor.predict()
函数。