From 436b6acc6ffedb29bd84e4b5d8f7c332760ac1f2 Mon Sep 17 00:00:00 2001 From: typhoonzero Date: Wed, 20 Sep 2017 16:09:48 +0800 Subject: [PATCH] follow comments --- paddle/operators/auc_op.cc | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/paddle/operators/auc_op.cc b/paddle/operators/auc_op.cc index 89f379b78f9..e7275a5933c 100644 --- a/paddle/operators/auc_op.cc +++ b/paddle/operators/auc_op.cc @@ -42,17 +42,17 @@ class AucOpMaker : public framework::OpProtoAndCheckerMaker { AucOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker) : OpProtoAndCheckerMaker(proto, op_checker) { AddInput("Inference", - "A floating point `Tensor` of arbitrary shape and whose values" - "are in the range `[0, 1]`."); + "A floating point tensor of arbitrary shape and whose values" + "are in the range [0, 1]."); AddInput("Label", - "A `Tensor` whose shape matches " - "`Inference`. Will be cast to `bool`."); + "A tensor whose shape matches " + "Inference. Will be cast to bool."); // TODO(typhoonzero): support weight input AddOutput("AUC", - "A scalar `Tensor` representing the " + "A scalar representing the " "current area-under-curve."); - AddAttr("curve", "Possible curves are ROC and PR") + AddAttr("curve", "Curve type, can be 'ROC' or 'PR'.") .SetDefault("ROC"); AddAttr("num_thresholds", "The number of thresholds to use when discretizing the" @@ -62,7 +62,8 @@ class AucOpMaker : public framework::OpProtoAndCheckerMaker { AddComment( R"DOC(Computes the AUC according forward output and label. Best to use for binary classification evaluations. - If `label` can be values other than 0 and 1, it will be cast + + If input label contains values other than 0 and 1, it will be cast to bool. You can find the definations here: -- GitLab