DomainModel

quantrs2_ml::domain_templates

Trait DomainModel 

Source 
pub trait DomainModel: Send + Sync {
    // Required methods
    fn name(&self) -> &str;
    fn domain(&self) -> Domain;
    fn predict(&self, input: &ArrayD<f64>) -> Result<ArrayD<f64>>;
    fn train(&mut self, X: &ArrayD<f64>, y: &ArrayD<f64>) -> Result<()>;
    fn config(&self) -> &TemplateConfig;
    fn training_suggestions(&self) -> Vec<String>;
    fn preprocessing_requirements(&self) -> Vec<String>;
}
Expand description

Trait for domain-specific models

Required Methods§

Source

fn name(&self) -> &str

Model name

Source

fn domain(&self) -> Domain

Domain

Source

fn predict(&self, input: &ArrayD<f64>) -> Result<ArrayD<f64>>

Make prediction

Source

fn train(&mut self, X: &ArrayD<f64>, y: &ArrayD<f64>) -> Result<()>

Train the model

Source

fn config(&self) -> &TemplateConfig

Get model configuration

Source

fn training_suggestions(&self) -> Vec<String>

Get training suggestions

Source

fn preprocessing_requirements(&self) -> Vec<String>

Get preprocessing requirements

Implementors§