Struct linfa::dataset::multi_target_model::MultiTargetModel

pub struct MultiTargetModel<R: Records, L> { /* fields omitted */ }

Implementations

impl<R: Records, L> MultiTargetModel<R, L>

pub fn new(models: Vec<Box<dyn PredictRef<R, Array1<L>>>>) -> Self

Create a wrapper model from a list of single-target models

The type parameter of the single-target models are only constraint to implement the prediction trait and can otherwise contain any object. This allows the mixture of different models into the same wrapper. If you want to use the same model for all predictions, just use the FromIterator implementation.

Trait Implementations

impl<F: Float, D: Data<Elem = F>, L, P: PredictRef<ArrayBase<D, Ix2>, Array1<L>> + 'static> FromIterator<P> for MultiTargetModel<ArrayBase<D, Ix2>, L>

fn from_iter<I: IntoIterator<Item = P>>(iter: I) -> Self

Creates a value from an iterator.

impl<L, F: Float, D: Data<Elem = F>> PredictRef<ArrayBase<D, Dim<[usize; 2]>>, ArrayBase<OwnedRepr<L>, Dim<[usize; 2]>>> for MultiTargetModel<ArrayBase<D, Ix2>, L>

fn predict_ref(&self, arr: &ArrayBase<D, Ix2>) -> Array2<L>