Trait rusty_machine::learning::SupModel

pub trait SupModel<T, U> {
    fn predict(&self, inputs: &T) -> U;
    fn train(&mut self, inputs: &T, targets: &U);
}

Trait for supervised model.

Required Methods

fn predict(&self, inputs: &T) -> U

Predict output from inputs.

fn train(&mut self, inputs: &T, targets: &U)

Train the model using inputs and targets.

Implementors

• impl<C: Criterion> SupModel<Matrix<f64>, Vector<f64>> for GenLinearModel<C>
• impl SupModel<Matrix<f64>, Vector<f64>> for LinRegressor
• impl<A> SupModel<Matrix<f64>, Vector<f64>> for LogisticRegressor<A> where A: OptimAlgorithm<BaseLogisticRegressor>
• impl<'a, T, A> SupModel<Matrix<f64>, Matrix<f64>> for NeuralNet<'a, T, A> where T: Criterion, A: OptimAlgorithm<BaseNeuralNet<'a, T>>
• impl<T: Kernel, U: MeanFunc> SupModel<Matrix<f64>, Vector<f64>> for GaussianProcess<T, U>
• impl<K: Kernel> SupModel<Matrix<f64>, Vector<f64>> for SVM<K>
• impl<T: Distribution> SupModel<Matrix<f64>, Matrix<f64>> for NaiveBayes<T>