tensorflow - 如何更改 tensorflow 0.12 版本的代码

Question

我有一个代码，它以前运行过，但现在没有。

# ## Create a regressor with TF Learn
# **Parameters**:
# - model_fn: regression model
# - n_classes: 0 for regression
# - verbose
# - steps: training steps
# - optimizer: ("SGD", "Adam", "Adagrad")
# - learning_rate
# - batch_size
    regressor = learn.TensorFlowEstimator(model_fn=lstm_model(TIMESTEPS, RNN_LAYERS, DENSE_LAYERS),n_classes=0,verbose=1,steps=TRAINING_STEPS,optimizer='Adagrad',learning_rate=0.03,batch_size=BATCH_SIZE)

但现在我收到错误消息"AttributeError: 'tensorflow.contrib.learn' object has no attribute 'TensorFlowEstimator'"

如何为 Tensorflow 0.12 修改此指令？

Answer 1

您应该替换TensorFlowEstimator为Estimator. 在 0.12 中进行了重大的重构，请查看Estimator文档。