summaryrefslogtreecommitdiff
path: root/candle-book
diff options
context:
space:
mode:
authorNicolas Patry <patry.nicolas@protonmail.com>2023-08-29 12:51:59 +0200
committerNicolas Patry <patry.nicolas@protonmail.com>2023-08-29 12:51:59 +0200
commit2d5b7a735d2c9ccb890dae73862dc734ef0950ae (patch)
tree7f089217f68f9f19d5af320af91475dd83f32989 /candle-book
parentd5a525f7a79c0ce7c7af4a235ec90796e3135b3b (diff)
downloadcandle-2d5b7a735d2c9ccb890dae73862dc734ef0950ae.tar.gz
candle-2d5b7a735d2c9ccb890dae73862dc734ef0950ae.tar.bz2
candle-2d5b7a735d2c9ccb890dae73862dc734ef0950ae.zip
Update the book with new layout of datasets.
Diffstat (limited to 'candle-book')
-rw-r--r--candle-book/src/lib.rs8
1 files changed, 4 insertions, 4 deletions
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