Skip to content

Commit 021fac6

Browse files
committed
Fix TFDS example
Looks like the record set name has changed also with the split names.
1 parent e6d8860 commit 021fac6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,12 +42,12 @@ print(mlc.Dataset(url).metadata.to_json())
4242
import tensorflow_datasets as tfds
4343
builder = tfds.core.dataset_builders.CroissantBuilder(
4444
jsonld=url,
45-
record_set_ids=["record_set_fashion_mnist"],
45+
record_set_ids=["fashion_mnist"],
4646
file_format='array_record',
4747
)
4848
builder.download_and_prepare()
4949
# 4. Split for training/testing
50-
train, test = builder.as_data_source(split=['default[:80%]', 'default[80%:]'])
50+
train, test = builder.as_data_source(split=['train', 'test'])
5151
```
5252

5353
Please see the [notebook recipes](python/mlcroissant/recipes) for more examples.

0 commit comments

Comments
 (0)