eegdino 0.2.0

EEG-DINO Foundation Model — inference in Rust with RLX
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

//! Convenience re-exports for common usage patterns.
//!
//! ```rust,ignore
//! use eegdino_rs::prelude::*;
//!
//! let device = parse_device("cpu")?;
//! let (mut enc, _) = EegDinoEncoder::load(path, None, device)?;
//! ```

pub use crate::config::{ModelConfig, ModelSize};
pub use crate::error::{EegDinoError, Result};
pub use crate::init_threads;

pub use crate::{
    detect_model_size, device_label, feature_for, is_device_available, parse_device,
    EegDinoEncoder, EegDinoEncoderBuilder, EncodingResult,
};

#[cfg(feature = "burn")]
pub use crate::inference::{
    ClassificationResult, EegDinoClassifier, EegDinoEncoder as BurnEegDinoEncoder,
    EegDinoEncoderBuilder as BurnEegDinoEncoderBuilder, EncodingResult as BurnEncodingResult,
};

#[cfg(feature = "burn")]
pub use crate::model::classifier::ClassificationModel;

#[cfg(feature = "burn")]
pub use crate::model::embedding::{EmbeddingCache, PatchEmbedding};

#[cfg(feature = "burn")]
pub use crate::model::encoder::EEGEncoder;