Browse Source

!80 Fix an API issue.

Merge pull request !80 from jxlang910/master
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
bc225f1574
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("..") sys.path.append("..")
from data_processing import generate_mnist_dataset from data_processing import generate_mnist_dataset
LOGGER = LogUtil.get_instance() LOGGER = LogUtil.get_instance()
LOGGER.set_level('INFO')

TAG = "Lenet5_train" 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 NoiseMechanismsFactory
from .mechanisms.mechanisms import ClipMechanismsFactory from .mechanisms.mechanisms import ClipMechanismsFactory
from .monitor.monitor import PrivacyMonitorFactory from .monitor.monitor import PrivacyMonitorFactory
from .monitor.monitor import RDPMonitor
from .monitor.monitor import ZCDPMonitor
from .optimizer.optimizer import DPOptimizerClassFactory from .optimizer.optimizer import DPOptimizerClassFactory
from .train.model import DPModel from .train.model import DPModel


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

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

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


Args: Args:
policy (str): Monitor policy, 'rdp' and 'zcdp' are supported 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 args (Union[int, float, numpy.ndarray, list, str]): Parameters
used for creating a privacy monitor. used for creating a privacy monitor.
kwargs (Union[int, float, numpy.ndarray, list, str]): Keyword kwargs (Union[int, float, numpy.ndarray, list, str]): Keyword


Loading…
Cancel
Save