| """ | |
| 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', | |
| ] |