FailedPreconditionError :: 2 erros de raiz encontrados.

Devido à inconsistência das versões do Tensorflow, o Estimator tem muitos poços! ! !

O código do problema é o seguinte

# 预定义estimator使用
output_dir = 'baseline_model'
if not os.path.exists(output_dir):
    os.mkdir(output_dir)
# tensorflow版本有问题,需要改!!!
baseline_estimator = tf.estimator.BaselineClassifier(model_dir=output_dir,
                                                    n_classes=2)
baseline_estimator.train(input_fn=lambda : make_dataset(
train_df,y_train,epochs=100))

A solução é a seguinte:

baseline_estimator = tf.compat.v1.estimator.BaselineClassifier(model_dir=output_dir,
                                                    n_classes=2)

 

Acho que você gosta

Origin blog.csdn.net/weixin_40539952/article/details/107855843
Recomendado
Clasificación