Add import of log_loss in keras_evaluation.py.

PiperOrigin-RevId: 468294581
This commit is contained in:
Steve Chien 2022-08-17 14:30:44 -07:00 committed by A. Unique TensorFlower
parent 052f9a3128
commit 5dd11fcdd6

View file

@ -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):