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
"""Trained models for membership inference attacks."""
from dataclasses import dataclass
import dataclasses
import numpy as np
from sklearn import ensemble
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
@dataclass
@dataclasses.dataclass
class AttackerData:
"""Input data for an ML classifier attack.