【ML&DL学习】14 tf premade estimator

预定义estimator使用

# 定义一个文件保存中间输出模型 
output_dir = 'baseline_model'
if not os.path.exists(output_dir):
    os.mkdir(output_dir)

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))

这个框架有bug 之后bug修复 在继续学习