def test_weight_column_should_not_be_used_as_feature(self): with self.assertRaisesRegexp(ValueError, 'weight_column should not be used as feature'): parsing_utils.regressor_parse_example_spec( feature_columns=[fc.numeric_column('a')], label_key='b', weight_column=fc.numeric_column('a'))
def test_weight_column_should_be_a_numeric_column(self): with self.assertRaisesRegexp(ValueError, 'tf.feature_column.numeric_column'): not_a_numeric_column = 3 parsing_utils.regressor_parse_example_spec( feature_columns=[fc.numeric_column('a')], label_key='b', weight_column=not_a_numeric_column)
def test_defaults(self): parsing_spec = parsing_utils.regressor_parse_example_spec( feature_columns=[fc.numeric_column('a')], label_key='b') expected_spec = { 'a': parsing_ops.FixedLenFeature((1,), dtype=dtypes.float32), 'b': parsing_ops.FixedLenFeature((1,), dtype=dtypes.float32), } self.assertDictEqual(expected_spec, parsing_spec)