未验证 提交 c6665795 编写于 作者: J jed 提交者: GitHub

Merge pull request #111 from kaih70/master

add auc
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
MPC Metrics MPC Metrics
""" """
import paddle.fluid.metrics
from paddle.fluid.metrics import MetricBase from paddle.fluid.metrics import MetricBase
import numpy as np import numpy as np
...@@ -23,6 +24,7 @@ import scipy ...@@ -23,6 +24,7 @@ import scipy
__all__ = [ __all__ = [
'KSstatistic', 'KSstatistic',
'Auc',
] ]
...@@ -108,3 +110,6 @@ class KSstatistic(MetricBase): ...@@ -108,3 +110,6 @@ class KSstatistic(MetricBase):
""" """
return scipy.stats.ks_2samp(self._data1, self._data2).statistic return scipy.stats.ks_2samp(self._data1, self._data2).statistic
Auc = paddle.fluid.metrics.Auc
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册