burn
burn copied to clipboard
Add `DataframeDataset` example
Hi,
The dataframe dataset has already been merged into burn and is demonstrated in burns book:
But I cannot find it in burn's docs and examples, where is it ?
Hello,
The code is separated in differents crates. You Can find the dataset crate under the crates folder: https://github.com/tracel-ai/burn/tree/main/crates%2Fburn-dataset.
You can find an example of the InMemDataset:
- InMemDataset: https://github.com/tracel-ai/burn/tree/main/examples/custom-csv-dataset
I don't know if there are more examples
Right now there are no examples which make use of the DataframeDataset (which is under the dataframe feature flag for the dataset module).
I agree that it would be nice to have at least a snippet for its usage. I'll convert this issue to track this addition.
Yeah it'd be great if we have example. Meanwhile you can see some basic tests:
https://github.com/tracel-ai/burn/blob/main/crates/burn-dataset/src/dataset/dataframe.rs
BTW, the API is available via a dataframe feature flag.