Struct CrossModalContrastive 

pub struct CrossModalContrastive<S = Untrained> { /* private fields */ }

Cross-modal contrastive learning model

Implementations

impl CrossModalContrastive<Untrained>

pub fn new() -> Self

Create a new cross-modal contrastive learning model

pub fn projection_dim(self, dim: usize) -> Self

Set projection dimension

pub fn hidden_dims(self, dims: Vec<usize>) -> Self

Set hidden dimensions

pub fn temperature(self, temp: f64) -> Self

Set temperature for contrastive loss

pub fn learning_rate(self, lr: f64) -> Self

Set learning rate

pub fn max_iter(self, max_iter: usize) -> Self

Set maximum iterations

pub fn contrastive_weight(self, weight: f64) -> Self

Set contrastive loss weight

pub fn supervised_weight(self, weight: f64) -> Self

Set supervised loss weight

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

Set random state

impl CrossModalContrastive<CrossModalContrastiveTrained>

pub fn get_embeddings( &self, x1: &ArrayView2<'_, f64>, x2: &ArrayView2<'_, f64>, ) -> SklResult<(Array2<f64>, Array2<f64>)>

Get embeddings for both modalities (trained model)

Trait Implementations

impl<S: Clone> Clone for CrossModalContrastive<S>

fn clone(&self) -> CrossModalContrastive<S>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<S: Debug> Debug for CrossModalContrastive<S>

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

Formats the value using the given formatter.

impl Default for CrossModalContrastive<Untrained>

fn default() -> Self

Returns the “default value” for a type.

impl Estimator for CrossModalContrastive<Untrained>

type Config = ()

Configuration type for the estimator

type Error = SklearsError

Error type for the estimator

type Float = f64

The numeric type used by this estimator

fn config(&self) -> &Self::Config

Get estimator configuration

fn validate_config(&self) -> Result<(), SklearsError>

Validate estimator configuration with detailed error context

fn check_compatibility( &self, n_samples: usize, n_features: usize, ) -> Result<(), SklearsError>

Check if estimator is compatible with given data dimensions

fn metadata(&self) -> EstimatorMetadata

Get estimator metadata

impl Fit<(ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>), ArrayBase<ViewRepr<&i32>, Dim<[usize; 1]>>> for CrossModalContrastive<Untrained>

type Fitted = CrossModalContrastive<CrossModalContrastiveTrained>

The fitted model type

fn fit( self, input: &CrossModalInput, y: &ArrayView1<'_, i32>, ) -> SklResult<Self::Fitted>

Fit the model to the provided data with validation

fn fit_with_validation( self, x: &X, y: &Y, _x_val: Option<&X>, _y_val: Option<&Y>, ) -> Result<(Self::Fitted, FitMetrics), SklearsError>

where Self: Sized,

Fit with custom validation and early stopping

impl Predict<(ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>), ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>>> for CrossModalContrastive<CrossModalContrastiveTrained>

fn predict(&self, input: &CrossModalInput) -> SklResult<Array1<i32>>

Make predictions on the provided data

fn predict_with_uncertainty( &self, x: &X, ) -> Result<(Output, UncertaintyMeasure), SklearsError>

Make predictions with confidence intervals

impl PredictProba<(ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>), ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>> for CrossModalContrastive<CrossModalContrastiveTrained>

fn predict_proba(&self, input: &CrossModalInput) -> SklResult<Array2<f64>>

Predict class probabilities