Model Maker allow core dataset library to handle datasets with unknown sizes.

PiperOrigin-RevId: 547268411
2023-07-11 12:47:37 -07:00
parent 4788fddde9
commit 56bc019819
11 changed files with 68 additions and 45 deletions
@@ -85,4 +85,5 @@ class Dataset(classification_dataset.ClassificationDataset):
    text_label_ds = tf.data.Dataset.zip((text_ds, label_index_ds))

    return Dataset(
-        dataset=text_label_ds, size=len(texts), label_names=label_names)
+        dataset=text_label_ds, label_names=label_names, size=len(texts)
+    )