Browse Source

Fix api issues

tags/v0.7.0-beta
jin-xiulang 5 years ago
parent
commit
b4d148aa3f
3 changed files with 12 additions and 1 deletions
  1. +2
    -0
      example/mnist_demo/mnist_train.py
  2. +4
    -0
      mindarmour/diff_privacy/__init__.py
  3. +6
    -1
      mindarmour/diff_privacy/monitor/monitor.py

+ 2
- 0
example/mnist_demo/mnist_train.py View File

@@ -27,6 +27,8 @@ from mindarmour.utils.logger import LogUtil
sys.path.append("..")
from data_processing import generate_mnist_dataset
LOGGER = LogUtil.get_instance()
LOGGER.set_level('INFO')

TAG = "Lenet5_train"




+ 4
- 0
mindarmour/diff_privacy/__init__.py View File

@@ -7,6 +7,8 @@ from .mechanisms.mechanisms import AdaClippingWithGaussianRandom
from .mechanisms.mechanisms import NoiseMechanismsFactory
from .mechanisms.mechanisms import ClipMechanismsFactory
from .monitor.monitor import PrivacyMonitorFactory
from .monitor.monitor import RDPMonitor
from .monitor.monitor import ZCDPMonitor
from .optimizer.optimizer import DPOptimizerClassFactory
from .train.model import DPModel

@@ -16,5 +18,7 @@ __all__ = ['NoiseGaussianRandom',
'NoiseMechanismsFactory',
'ClipMechanismsFactory',
'PrivacyMonitorFactory',
'RDPMonitor',
'ZCDPMonitor',
'DPOptimizerClassFactory',
'DPModel']

+ 6
- 1
mindarmour/diff_privacy/monitor/monitor.py View File

@@ -40,7 +40,12 @@ class PrivacyMonitorFactory:

Args:
policy (str): Monitor policy, 'rdp' and 'zcdp' are supported
by now.
by now. If policy is 'rdp', the monitor will compute the
privacy budget of DP training based on Renyi differential
privacy theory; If policy is 'zcdp', the monitor will compute
the privacy budget of DP training based on zero-concentrated
differential privacy theory. It's worth noting that 'zcdp'
is not suitable for subsampling nosie mechanism.
args (Union[int, float, numpy.ndarray, list, str]): Parameters
used for creating a privacy monitor.
kwargs (Union[int, float, numpy.ndarray, list, str]): Keyword


Loading…
Cancel
Save