def test_get_config(self, gamma, pos_weight, from_logits, label_smoothing): """Check the get_config() method.""" loss1 = BinaryFocalLoss(gamma=gamma, pos_weight=pos_weight, from_logits=from_logits, label_smoothing=label_smoothing, name='binary_focal_loss') config1 = loss1.get_config() loss2 = BinaryFocalLoss(**config1) config2 = loss2.get_config() self.assertEqual(config1, config2)
def test_get_config(): """Check the get_config() method.""" for gamma, pos_weight in product((0, 1, 2), (None, 0.5)): for from_logits, label_smoothing in product((True, False), (None, 0.1)): loss1 = BinaryFocalLoss(gamma=gamma, pos_weight=pos_weight, from_logits=from_logits, label_smoothing=label_smoothing, name='binary_focal_loss') config1 = loss1.get_config() loss2 = BinaryFocalLoss(**config1) config2 = loss2.get_config() assert config1 == config2