小编Ash*_*man的帖子

Tensorflow估计器:切换到autumn_interpolation以获取模型的正确PR-AUC

在我的项目中,我使用的是预制估算器DNNClassifier。这是我的估算器:

model = tf.estimator.DNNClassifier(
        hidden_units=network,
        feature_columns=feature_cols,
        n_classes= 2,
        activation_fn=tf.nn.relu,
        optimizer=tf.train.ProximalAdagradOptimizer(
            learning_rate=0.1,
            l1_regularization_strength=0.001
        ),
        config=chk_point_run_config,
        model_dir=MODEL_CHECKPOINT_DIR
    )
Run Code Online (Sandbox Code Playgroud)

当我使用评估模型时eval_res = model.evaluate(..),出现以下警告:

警告:tensorflow:梯形规则会产生不正确的PR-AUC。请改用“ careful_interpolation”。

我如何才能切换到autumn_interpolation以从该evaluate()方法获得正确的结果?

Tensorflow版本: 1.8

python tensorflow

3
推荐指数
1
解决办法
2001
查看次数

标签 统计

python ×1

tensorflow ×1