File tree Expand file tree Collapse file tree 1 file changed +1
-3
lines changed
art/attacks/inference/membership_inference Expand file tree Collapse file tree 1 file changed +1
-3
lines changed Original file line number Diff line number Diff line change @@ -60,7 +60,7 @@ class MembershipInferenceBlackBox(MembershipInferenceAttack):
60
60
"scaler_type" ,
61
61
"nn_model_epochs" ,
62
62
"nn_model_batch_size" ,
63
- "nn_model_learning_rate"
63
+ "nn_model_learning_rate" ,
64
64
]
65
65
_estimator_requirements = (BaseEstimator , (ClassifierMixin , RegressorMixin ))
66
66
@@ -256,7 +256,6 @@ def fit( # pylint: disable=W0613
256
256
if x_2 is None :
257
257
self .use_label = False
258
258
259
- scaler : Optional [Any ] = None
260
259
if self .scaler_type :
261
260
if self .scaler_type == "standard" :
262
261
self .scaler = StandardScaler ()
@@ -267,7 +266,6 @@ def fit( # pylint: disable=W0613
267
266
else :
268
267
raise ValueError ("Illegal scaler_type: " , self .scaler_type )
269
268
270
-
271
269
if self .default_model and self .attack_model_type == "nn" :
272
270
import torch
273
271
from torch import nn
You can’t perform that action at this time.
0 commit comments