Struct FittedSupervisedContrastiveLearning 

pub struct FittedSupervisedContrastiveLearning {
    pub base_model: SupervisedContrastiveLearning,
    pub encoder_weights: Array2<f64>,
    pub classes: Array1<i32>,
    pub n_classes: usize,
    pub class_centroids: Array2<f64>,
}

Fitted Supervised Contrastive Learning model

Fields

base_model: SupervisedContrastiveLearning

base_model

encoder_weights: Array2<f64>

encoder_weights

classes: Array1<i32>

classes

n_classes: usize

n_classes

class_centroids: Array2<f64>

class_centroids

Trait Implementations

impl Clone for FittedSupervisedContrastiveLearning

fn clone(&self) -> FittedSupervisedContrastiveLearning

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FittedSupervisedContrastiveLearning

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

Formats the value using the given formatter.

impl Predict<ArrayBase<ViewRepr<&f64>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<i32>, Dim<[usize; 1]>>> for FittedSupervisedContrastiveLearning

fn predict(&self, X: &ArrayView2<'_, f64>) -> Result<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<ViewRepr<&f64>, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<f64>, Dim<[usize; 2]>>> for FittedSupervisedContrastiveLearning

fn predict_proba(&self, X: &ArrayView2<'_, f64>) -> Result<Array2<f64>>

Predict class probabilities