626_privacy/tensorflow_privacy
2
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

Fix TensorFlow Privacy g-importing-member lint error.

Browse source 
PiperOrigin-RevId: 424356142
This commit is contained in:
Michael Reneer 2022-01-26 08:40:29 -08:00 committed by A. Unique TensorFlower
parent 4b76e882bc
commit b309916927
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tensorflow_privacy/privacy/privacy_tests/membership_inference_attack/models.py
View file

@ -15,7 +15,8 @@
# Lint as: python3 # Lint as: python3
"""Trained models for membership inference attacks.""" """Trained models for membership inference attacks."""
from dataclasses import dataclass import dataclasses
import numpy as np import numpy as np
from sklearn import ensemble from sklearn import ensemble
from sklearn import linear_model from sklearn import linear_model
@ -27,7 +28,7 @@ from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_s
from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import DataSize from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import DataSize
@dataclass @dataclasses.dataclass
class AttackerData: class AttackerData:
"""Input data for an ML classifier attack. """Input data for an ML classifier attack.