def test_dense_feature_scaler(self):
tensor = tf.constant([-15., -12., -20., 5., -100.])
expected = [0, 0.6, -1., 1., -1.]
scaler = run_finetuning_lib.DenseFeatureScaler(min_value=-20,
max_value=-10)
self.assertAllClose(expected, scaler.transform(tensor))
def test_dense_feature_scaler_invalid_range(self):
with self.assertRaises(ValueError):
run_finetuning_lib.DenseFeatureScaler(min_value=10, max_value=5)
with self.assertRaises(ValueError):
run_finetuning_lib.DenseFeatureScaler(min_value=10, max_value=10)