FailedPreconditionError:: 2 root error(s) found.

Due to the inconsistency of Tensorflow versions, Estimator has many pits! ! !

The problem code is as follows

# 预定义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))

The solution is as follows:

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

 

Guess you like

Origin blog.csdn.net/weixin_40539952/article/details/107855843