diff --git a/mindarmour/diff_privacy/__init__.py b/mindarmour/diff_privacy/__init__.py
index a54ce80b771d621d6b61b942688dcf1e1d0f8ece..6dbd1e2420cb4726c60242c91444f05267bd5635 100644
--- a/mindarmour/diff_privacy/__init__.py
+++ b/mindarmour/diff_privacy/__init__.py
@@ -11,6 +11,7 @@ from .monitor.monitor import RDPMonitor
 from .monitor.monitor import ZCDPMonitor
 from .optimizer.optimizer import DPOptimizerClassFactory
 from .train.model import DPModel
+from .evaluation.membership_inference import MembershipInference
 
 __all__ = ['NoiseGaussianRandom',
            'NoiseAdaGaussianRandom',
@@ -21,4 +22,5 @@ __all__ = ['NoiseGaussianRandom',
            'RDPMonitor',
            'ZCDPMonitor',
            'DPOptimizerClassFactory',
-           'DPModel']
+           'DPModel',
+           'MembershipInference']