File size: 779 Bytes
6640531 dbbc4dd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | """
Training module for DP-SGD Explorer.
Contains:
- MockTrainer: Simulation-based training for fast experimentation
- SimplifiedRealTrainer: Real TensorFlow-based DP-SGD training
- RealTrainer: Full TensorFlow Privacy-based DP-SGD training
- PrivacyCalculator: Unified RDP-based privacy accounting
- gradient_utils: Shared gradient visualization utilities
"""
from .mock_trainer import MockTrainer
from .privacy_calculator import PrivacyCalculator, get_privacy_calculator
from .gradient_utils import (
generate_gradient_norms,
generate_clipped_gradients,
generate_gradient_info
)
__all__ = [
'MockTrainer',
'PrivacyCalculator',
'get_privacy_calculator',
'generate_gradient_norms',
'generate_clipped_gradients',
'generate_gradient_info',
] |