dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/examples/custom-csv-dataset
History
Sylvain Benner 4427768570
[refactor] Move burn crates to their own crates directory (#1336) 		2024-02-20 13:57:55 -05:00
..
examples Add example for custom CSV dataset (#1129) 2024-01-11 09:24:25 -05:00
src Add example for custom CSV dataset (#1129) 2024-01-11 09:24:25 -05:00
.gitignore Add example for custom CSV dataset (#1129) 2024-01-11 09:24:25 -05:00
Cargo.toml [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
README.md Add example for custom CSV dataset (#1129) 2024-01-11 09:24:25 -05:00

README.md

Custom CSV Dataset

The custom-csv-dataset example implements the Dataset trait to retrieve dataset elements from a .csv file on disk. For this example, we use the diabetes dataset (original source).

The dataset only contains 442 records, so we use InMemDataset::from_csv(path) method to read the csv dataset file into a vector (in-memory) of DiabetesPatient records (struct) with the help of serde.

Example Usage

cargo run --example custom-csv-dataset