burn/examples/custom-csv-dataset
Nathaniel Simard d50bac165e
feat cube support Array (#1907)
2024-06-19 17:03:02 -04: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 feat cube support Array (#1907) 2024-06-19 17:03:02 -04: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