search

tracel-ai / burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.
https://burn.dev
Apache License 2.0
8.95k stars 444 forks source link

Add `DataframeDataset` example #2406

Open wangjiawen2013 opened 3 weeks ago

wangjiawen2013 commented 3 weeks ago

Hi, The dataframe dataset has already been merged into burn and is demonstrated in burns book: image

But I cannot find it in burn's docs and examples, where is it ?

HessTaha commented 3 weeks ago

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:

I don't know if there are more examples

laggui commented 3 weeks ago

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.

antimora commented 2 weeks ago

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

antimora commented 2 weeks ago

BTW, the API is available via a dataframe feature flag.