diff --git a/python/paddle_fl/mpc/metrics.py b/python/paddle_fl/mpc/metrics.py index 57bfeac618e98da7826156bc05c4d87655fcaca6..df656e1bb08d36f6e2b8f983f4fc06c099f4fb39 100644 --- a/python/paddle_fl/mpc/metrics.py +++ b/python/paddle_fl/mpc/metrics.py @@ -15,6 +15,7 @@ MPC Metrics """ +import paddle.fluid.metrics from paddle.fluid.metrics import MetricBase import numpy as np @@ -23,6 +24,7 @@ import scipy __all__ = [ 'KSstatistic', + 'Auc', ] @@ -108,3 +110,6 @@ class KSstatistic(MetricBase): """ return scipy.stats.ks_2samp(self._data1, self._data2).statistic + + +Auc = paddle.fluid.metrics.Auc