Trait linfa::traits::Predict

pub trait Predict<R: Records, T> {
    // Required method
    fn predict(&self, x: R) -> T;
}

Predict with model

This trait assumes the PredictInplace implementation and provides additional input/output combinations.

Provided implementation

use linfa::traits::Predict;

// predict targets with reference to dataset (&Dataset -> Array)
let pred_targets = model.predict(&dataset);
// predict targets inside dataset (Dataset -> Dataset)
let pred_dataset = model.predict(dataset);
// or use a record datastruct directly (Array -> Dataset)
let pred_targets = model.predict(x);

Required Methods

fn predict(&self, x: R) -> T

Implementors

impl<'a, F, D, DM, T, O> Predict<&'a ArrayBase<D, DM>, T> for O

where D: Data<Elem = F>, DM: Dimension, O: PredictInplace<ArrayBase<D, DM>, T>,

impl<'a, F, R, T, S, O> Predict<&'a DatasetBase<R, T>, S> for O

where R: Records<Elem = F>, O: PredictInplace<R, S>,

impl<F, D, E, T, O> Predict<ArrayBase<D, Dim<[usize; 2]>>, DatasetBase<ArrayBase<D, Dim<[usize; 2]>>, T>> for O

where D: Data<Elem = F>, T: AsTargets<Elem = E>, O: PredictInplace<ArrayBase<D, Ix2>, T>,

impl<F, R, T, E, S, O> Predict<DatasetBase<R, T>, DatasetBase<R, S>> for O

where R: Records<Elem = F>, S: AsTargets<Elem = E>, O: PredictInplace<R, S>,