burn/examples/custom-csv-dataset
Guillaume Lagrange 6079f98950
Fix missing ticks in Burn book and remove unused example dependency (#1144)
* Add missing ticks for triu and tril ops

* Removed example unused dependency
2024-01-16 09:03:17 -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 Fix missing ticks in Burn book and remove unused example dependency (#1144) 2024-01-16 09:03:17 -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