コード例 #1
0
def set_protobuf_float(column_schema, feature):
    domain = column_schema.properties.get("domain", {})
    feature.float_domain.CopyFrom(
        schema_pb2.FloatDomain(
            name=column_schema.name,
            min=domain.get("min", None),
            max=domain.get("max", None),
        ))
    feature.type = schema_pb2.FeatureType.FLOAT
    return feature
コード例 #2
0
         'x': schema_pb2.StringDomain(value=['a', 'b'])
     }
 },
 {
     'testcase_name': 'float_domain',
     'ascii_proto': """
       feature: {
         name: "x" type: FLOAT
         float_domain {min: 0.0 max: 0.5}
       }
     """,
     'feature_spec': {
         'x': tf.io.VarLenFeature(tf.float32)
     },
     'domains': {
         'x': schema_pb2.FloatDomain(min=0.0, max=0.5)
     }
 },
 {
     'testcase_name': 'sparse_feature_rank_0',
     'ascii_proto': """
       feature {
         name: "value_key"
         type: INT
       }
       sparse_feature {
         name: "x"
         value_feature {name: "value_key"}
       }
     """,
     'feature_spec': {
コード例 #3
0
    'feature_name':
    'x',
    'domain':
    schema_pb2.IntDomain(min=1, max=5),
    'output_schema_proto_text':
    '''
          feature { name: 'x' int_domain { min: 1 max: 5 } }'''
}, {
    'testcase_name':
    'float_domain',
    'input_schema_proto_text':
    '''feature { name: 'x' }''',
    'feature_name':
    'x',
    'domain':
    schema_pb2.FloatDomain(min=1.1, max=5.1),
    'output_schema_proto_text':
    '''
          feature { name: 'x' float_domain { min: 1.1 max: 5.1 } }'''
}, {
    'testcase_name':
    'string_domain',
    'input_schema_proto_text':
    '''feature { name: 'x' }''',
    'feature_name':
    'x',
    'domain':
    schema_pb2.StringDomain(value=['a', 'b']),
    'output_schema_proto_text':
    '''
          feature { name: 'x' string_domain { value: 'a' value: 'b' } }'''
コード例 #4
0
     },
     'exception_type': ValueError,
     'error_message': 'Invalid semantic_domain_stats_sample_rate 2'
 },
 {
     'testcase_name':
     'categorical_float_without_sketch_generators',
     'stats_options_kwargs': {
         'experimental_use_sketch_based_topk_uniques':
         False,
         'schema':
         schema_pb2.Schema(feature=[
             schema_pb2.Feature(
                 name='f',
                 type=schema_pb2.FLOAT,
                 float_domain=schema_pb2.FloatDomain(is_categorical=True))
         ], ),
     },
     'exception_type':
     ValueError,
     'error_message': ('Categorical float features set in schema require '
                       'experimental_use_sketch_based_topk_uniques'),
 },
 {
     'testcase_name': 'both_slice_fns_and_slice_sqls_specified',
     'stats_options_kwargs': {
         'experimental_slice_functions': [lambda x: (None, x)],
         'experimental_slice_sqls': ['']
     },
     'exception_type': ValueError,
     'error_message': 'Only one of experimental_slice_functions or'