metrics.rst 2.5 KB
Newer Older
D
dzhwinter 已提交
1 2 3 4
############
模型评估
############

5
模型评估是指用评价函数(metrics)来评估模型的好坏,可作为在训练中调整超参数、评估模型效果的重要依据。不同类型的模型任务会选取不同评价函数,常见的如回归类任务会用均方差(MSE),二分类任务会用AUC (Area Under Curve)值等。
D
dzhwinter 已提交
6

7 8 9 10 11 12
评价函数和loss函数非常相似,但不参与模型的训练优化。
 

评价函数的输入为模型的预测值(preds)和标注值(labels),并返回计算后的评价指标。

paddle.fluid.metrics模块提供了一系列常用的模型评价指标; 用户也可以通过Python接口定制评价指标,或者通过定制C++ Operator的方式,在GPU上加速评价指标的计算。
D
dzhwinter 已提交
13 14 15 16

常用指标
############

17 18 19 20 21 22 23
不同类型的任务,会选用不同的评价指标。
 
回归问题通常会用RMSE(均方根误差),MAE(平均绝对误差),R-Square(R平方)等
AUC(Area Under Cure)指标则常被用在分类任务(classification)上
目标检测任务(Object Detection)则经常会用到mAP(Mean Average Precision) 
 
paddle.fluid.metrics中包含了一些常用分类指标,例如Precision, Recall, Accuracy等 
D
dzhwinter 已提交
24

25
下面是使用Precision指标的示例:
D
dzhwinter 已提交
26 27 28

.. code-block:: python

29 30 31 32 33 34 35 36 37 38 39 40
   import paddle.fluid as fluid
   label = fluid.layers.data(name="label", shape=[1], dtype="int32")
   data = fluid.layers.data(name="data", shape=[32, 32], dtype="int32")
   pred = fluid.layers.fc(input=data, size=1000, act="tanh")
   acc = fluid.metrics.Precision()
   for pass_iter in range(PASSES):
     acc.reset()
     for data in train_reader():
         loss, preds, labels = exe.run(fetch_list=[cost, pred, label])
         acc.update(preds=preds, labels=labels)
     numpy_acc = acc.eval()

D
dzhwinter 已提交
41 42 43

自定义指标
############
44 45 46
Fluid支持自定义指标,可灵活支持各类计算任务。下面是一个自定义的简单计数器评价函数示例:

其中preds是模型预测值,labels是标注值。
D
dzhwinter 已提交
47 48 49

.. code-block:: python

50 51 52 53
   class MyMetric(MetricBase):
       def __init__(self, name=None):
           super(MyMetric, self).__init__(name)
           self.counter = 0  # simple counter
D
dzhwinter 已提交
54

55 56
       def reset(self):
           self.counter = 0
D
dzhwinter 已提交
57

58 59 60 61 62 63
       def update(self, preds, labels):
           if not _is_numpy_(preds):
               raise ValueError("The 'preds' must be a numpy ndarray.")
           if not _is_numpy_(labels):
               raise ValueError("The 'labels' must be a numpy ndarray.")
           self.counter += sum(preds == labels)
D
dzhwinter 已提交
64

65 66
       def eval(self):
           return self.counter