Struct FittedSimCLR 

pub struct FittedSimCLR {
    pub base_model: SimCLR,
    pub encoder_weights: Array2<f64>,
    pub projection_weights: Array2<f64>,
    pub classes: Array1<i32>,
    pub n_classes: usize,
}

Fitted SimCLR model

Fields

base_model: SimCLR

base_model

encoder_weights: Array2<f64>

encoder_weights

projection_weights: Array2<f64>

projection_weights

classes: Array1<i32>

classes

n_classes: usize

n_classes

Trait Implementations

impl Clone for FittedSimCLR

fn clone(&self) -> FittedSimCLR

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FittedSimCLR

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 FittedSimCLR

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 FittedSimCLR

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

Predict class probabilities