Struct DeviceModel 

pub struct DeviceModel<'a, M> { /* private fields */ }

Binds an accelerated model to a device while retaining the ordinary Predict interface consumed by every explainer.

Implementations

impl<'a, M> DeviceModel<'a, M>

pub fn new(model: &'a M, device: ExecutionDevice) -> Self

pub fn device(&self) -> ExecutionDevice

Trait Implementations

impl<M: AcceleratedPredict> Predict for DeviceModel<'_, M>

fn predict(&self, x: ArrayView2<'_, f64>) -> Result<Array2<f64>>

Predict outputs for a batch of samples.

fn predict_owned(&self, x: Array2<f64>) -> Result<Array2<f64>>

Predicts from an owned batch.

fn n_features(&self) -> Option<usize>

Returns the number of input features expected by the model.

fn n_outputs(&self) -> Option<usize>

Returns the number of outputs produced by the model.