burn_dataset/
lib.rs

1#![warn(missing_docs)]
2#![cfg_attr(docsrs, feature(doc_auto_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 dirs;
12
13/// Sources for datasets.
14pub mod source;
15
16/// Transformations to be used with datasets.
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
27mod dataset;
28pub use dataset::*;
29#[cfg(any(feature = "sqlite", feature = "sqlite-bundled"))]
30pub use source::huggingface::downloader::*;
31
32#[cfg(test)]
33mod test_data {
34    pub fn string_items() -> Vec<String> {
35        vec![
36            "1 Item".to_string(),
37            "2 Items".to_string(),
38            "3 Items".to_string(),
39            "4 Items".to_string(),
40        ]
41    }
42}