Struct catboost2::Model

pub struct Model { /* private fields */ }

Safe wrapper around a raw catboost model.

This is the central type of this library and is the entry point for any use of this library.

Implementations

impl Model

pub fn load<P: AsRef<Path>>(path: P) -> Result<Self>

Load a model from a file.

pub fn load_buffer<P: AsRef<Vec<u8>>>(buffer: P) -> Result<Self>

Load a model from a buffer.

pub fn calc_model_prediction(
    &self,
    float_features: Vec<Vec<f32>>,
    cat_features: Vec<Vec<String>>
) -> Result<Vec<f64>>

Calculate raw model predictions on float features and string categorical feature values

pub fn get_float_features_count(&self) -> u64

Get expected float feature count for model

pub fn get_cat_features_count(&self) -> u64

Get expected categorical feature count for model

pub fn get_tree_count(&self) -> u64

Get number of trees in model

pub fn get_dimensions_count(&self) -> u64

Get number of dimensions in model

Trait Implementations

impl Debug for Model

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Drop for Model

fn drop(&mut self)

Executes the destructor for this type.