Как я могу изменить код для версии tenorflow 0.12
У меня есть код, который работал раньше, но не сейчас.
# ## 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?
1 ответ
Вы должны заменить TensorFlowEstimator
с Estimator
, В 0,12 произошел существенный перефактор, пожалуйста, посмотрите на Estimator
Docs.