예제 #1
0
 def _info(self):
   return dataset_info.DatasetInfo(
       builder=self,
       features=features.FeaturesDict({
           "image": features.Image(shape=(16, 16, 1)),
           "label": features.ClassLabel(names=["dog", "cat"]),
           "id": tf.int32,
       }),
       supervised_keys=("x", "x"),
       metadata=dataset_info.BeamMetadataDict(),
   )
 def _info(self):
     return dataset_info.DatasetInfo(
         builder=self,
         features=features.FeaturesDict({
             'image':
             features.Image(shape=(16, 16, 1)),
             'label':
             features.ClassLabel(names=['dog', 'cat']),
             'id':
             tf.int32,
         }),
         supervised_keys=('x', 'x'),
         metadata=dataset_info.BeamMetadataDict(),
     )