Struct UMAP 

pub struct UMAP { /* private fields */ }

UMAP (Uniform Manifold Approximation and Projection) dimensionality reduction

UMAP constructs a high dimensional graph representation of the data then optimizes a low dimensional graph to be as structurally similar as possible.

Implementations

impl UMAP

pub fn new( n_neighbors: usize, n_components: usize, mindist: f64, learning_rate: f64, n_epochs: usize, ) -> Self

Creates a new UMAP instance

Arguments

• n_neighbors - Number of neighbors to consider for local structure (default: 15)
• n_components - Number of dimensions in the low dimensional space (default: 2)
• mindist - Minimum distance between points in low dimensional space (default: 0.1)
• learning_rate - Learning rate for optimization (default: 1.0)
• n_epochs - Number of epochs for optimization (default: 200)

pub fn with_random_state(self, seed: u64) -> Self

Sets the random state for reproducibility

pub fn with_metric(self, metric: &str) -> Self

Sets the distance metric

pub fn fit<S>(&mut self, x: &ArrayBase<S, Ix2>) -> Result<()>

where S: Data, S::Elem: Float + NumCast + Send + Sync,

Fits the UMAP model to the input data

Arguments

• x - The input data, shape (nsamples, n_features)

Returns

• Result<()> - Ok if successful, Err otherwise

pub fn transform<S>(&self, x: &ArrayBase<S, Ix2>) -> Result<Array2<f64>>

where S: Data, S::Elem: Float + NumCast,

Transforms the input data using the fitted UMAP model

Arguments

• x - The input data, shape (nsamples, n_features)

Returns

• Result<Array2<f64>> - The transformed data, shape (nsamples, n_components)

pub fn fit_transform<S>(&mut self, x: &ArrayBase<S, Ix2>) -> Result<Array2<f64>>

where S: Data, S::Elem: Float + NumCast + Send + Sync,

Fits the UMAP model to the input data and returns the embedding

Arguments

• x - The input data, shape (nsamples, n_features)

Returns

• Result<Array2<f64>> - The transformed data, shape (nsamples, n_components)

pub fn embedding(&self) -> Option<&Array2<f64>>

Returns the low dimensional embedding

Trait Implementations

impl Clone for UMAP

fn clone(&self) -> UMAP

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for UMAP

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.