Esempi in Python per _tfds_filenames

Esempio n. 1

File: audio_to_embeddings_beam_utils_test.py Progetto: xjdrlabs/google-research

 def test_tfds_info(self, dataset_name):
   self.assertTrue(audio_to_embeddings_beam_utils._tfds_sample_rate(
       dataset_name))
   self.assertTrue(audio_to_embeddings_beam_utils._tfds_filenames(
       dataset_name, 'train'))
   self.assertTrue(audio_to_embeddings_beam_utils._tfds_filenames(
       dataset_name, 'validation'))
   self.assertTrue(audio_to_embeddings_beam_utils._tfds_filenames(
       dataset_name, 'test'))

Esempio n. 2

File: count_duration_beam.py Progetto: mengdong/google-research

 def _tfds_fns(ds_name):
     fns = [
         x  # pylint:disable=g-complex-comprehension
         for s in ('train', 'validation', 'test')
         for x in audio_to_embeddings_beam_utils._tfds_filenames(
             ds_name, s)
     ]  # pylint:disable=protected-access
     fns = [fns]  # TFRecords require a list.
     return (fns, 'tfrecord')