def test_class_weights(self): y = np.array([0, 1, 0, 0, 2]) class_weights = {0: 0.5, 1: 1.0, 2: 1.5} weights = training_utils_v1.standardize_weights( y, class_weight=class_weights ) self.assertAllClose(weights, np.array([0.5, 1.0, 0.5, 0.5, 1.5]))
def test_sample_weights_and_class_weights(self): y = np.array([0, 1, 0, 0, 2]) sample_weights = np.array([0.5, 1.0, 1.0, 0.0, 2.0]) class_weights = {0: 0.5, 1: 1.0, 2: 1.5} weights = training_utils_v1.standardize_weights( y, sample_weights, class_weights) expected = sample_weights * np.array([0.5, 1.0, 0.5, 0.5, 1.5]) self.assertAllClose(weights, expected)
def test_sample_weights(self): y = np.array([0, 1, 0, 0, 2]) sample_weights = np.array([0.5, 1.0, 1.0, 0.0, 2.0]) weights = training_utils_v1.standardize_weights(y, sample_weights) self.assertAllClose(weights, sample_weights)