Trait linfa::traits::Predict

pub trait Predict<R: Records, T> {
    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>,