From 2d5b7a735d2c9ccb890dae73862dc734ef0950ae Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Tue, 29 Aug 2023 12:51:59 +0200 Subject: Update the book with new layout of datasets. --- candle-book/src/lib.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'candle-book') diff --git a/candle-book/src/lib.rs b/candle-book/src/lib.rs index c89ddc94..ef9b853e 100644 --- a/candle-book/src/lib.rs +++ b/candle-book/src/lib.rs @@ -115,8 +115,8 @@ let repo = Repo::with_revision( "refs/convert/parquet".to_string(), ); let repo = api.repo(repo); -let test_parquet_filename = repo.get("mnist/mnist-test.parquet")?; -let train_parquet_filename = repo.get("mnist/mnist-train.parquet")?; +let test_parquet_filename = repo.get("mnist/test/0000.parquet")?; +let train_parquet_filename = repo.get("mnist/train/0000.parquet")?; let test_parquet = SerializedFileReader::new(std::fs::File::open(test_parquet_filename)?)?; let train_parquet = SerializedFileReader::new(std::fs::File::open(train_parquet_filename)?)?; // ANCHOR_END: book_training_1 @@ -129,8 +129,8 @@ for row in test_parquet { } } // ANCHOR_END: book_training_2 -let test_parquet_filename = repo.get("mnist/mnist-test.parquet")?; -let train_parquet_filename = repo.get("mnist/mnist-train.parquet")?; +let test_parquet_filename = repo.get("mnist/test/0000.parquet")?; +let train_parquet_filename = repo.get("mnist/train/0000.parquet")?; let test_parquet = SerializedFileReader::new(std::fs::File::open(test_parquet_filename)?)?; let train_parquet = SerializedFileReader::new(std::fs::File::open(train_parquet_filename)?)?; // ANCHOR: book_training_3 -- cgit v1.2.3