burn_dataset/
lib.rs

1#![warn(missing_docs)]
2#![cfg_attr(docsrs, feature(doc_cfg))]
3
4//! # Burn Dataset
5//!
6//! Burn Dataset is a library for creating and loading datasets.
7
8#[macro_use]
9extern crate derive_new;
10
11extern crate alloc;
12extern crate dirs;
13
14/// Sources for datasets.
15pub mod source;
16
17pub mod transform;
18
19/// Audio datasets.
20#[cfg(feature = "audio")]
21pub mod audio;
22
23/// Vision datasets.
24#[cfg(feature = "vision")]
25pub mod vision;
26
27/// Natural language processing datasets.
28#[cfg(feature = "nlp")]
29pub mod nlp;
30
31/// Network dataset utilities.
32#[cfg(feature = "network")]
33pub mod network {
34    pub use burn_std::network::*;
35}
36
37mod dataset;
38pub use dataset::*;
39#[cfg(any(feature = "sqlite", feature = "sqlite-bundled"))]
40pub use source::huggingface::downloader::*;
41
42#[cfg(test)]
43mod test_data {
44    pub fn string_items() -> Vec<String> {
45        vec![
46            "1 Item".to_string(),
47            "2 Items".to_string(),
48            "3 Items".to_string(),
49            "4 Items".to_string(),
50        ]
51    }
52}
burn_dataset/lib.rs

burn_dataset/
lib.rs
