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
31mod dataset;
32pub use dataset::*;
33#[cfg(any(feature = "sqlite", feature = "sqlite-bundled"))]
34pub use source::huggingface::downloader::*;
35
36#[cfg(test)]
37mod test_data {
38    pub fn string_items() -> Vec<String> {
39        vec![
40            "1 Item".to_string(),
41            "2 Items".to_string(),
42            "3 Items".to_string(),
43            "4 Items".to_string(),
44        ]
45    }
46}