forked from 626_privacy/tensorflow_privacy
Add import of log_loss in keras_evaluation.py.
PiperOrigin-RevId: 468294581
This commit is contained in:
parent
052f9a3128
commit
5dd11fcdd6
1 changed files with 1 additions and 0 deletions
|
@ -25,6 +25,7 @@ from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_s
|
|||
from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import get_flattened_attack_metrics
|
||||
from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SlicingSpec
|
||||
from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.utils_tensorboard import write_results_to_tensorboard_tf2 as write_results_to_tensorboard
|
||||
from tensorflow_privacy.privacy.privacy_tests.utils import log_loss
|
||||
|
||||
|
||||
def calculate_losses(model, data, labels, is_logit=False, batch_size=32):
|
||||
|
|
Loading…
Reference in a new issue