Struct EnsembleParams 

pub struct EnsembleParams { /* private fields */ }

Parameters for one ensemble-predict call — the sklearn-wrapper-equivalent counterpart to TabFMClassifier(...)/TabFMRegressor(...)’s constructor kwargs. Fields are private; build one by chaining .with_*() off EnsembleParams::default (defaults match the wrapper’s own constructor defaults).

use zsfm_tabfm::EnsembleParams;

let params = EnsembleParams::default()
    .with_n_estimators(64)
    .with_enable_nnls(true)
    .with_random_state(0);

Implementations

impl EnsembleParams

pub fn with_n_estimators(self, v: usize) -> Self

pub fn with_norm_methods(self, v: Vec<NormMethod>) -> Self

pub fn with_class_shift(self, v: bool) -> Self

pub fn with_outlier_threshold(self, v: f64) -> Self

pub fn with_softmax_temperature(self, v: f64) -> Self

pub fn with_average_logits(self, v: bool) -> Self

pub fn with_random_state(self, v: u64) -> Self

pub fn with_binary_calibration(self, v: bool) -> Self

Enable Platt scaling for binary classification (n_classes == 2).

pub fn with_multiclass_calibration(self, v: bool) -> Self

Enable vector scaling for multiclass classification (n_classes > 2).

pub fn with_num_folds_for_cv(self, v: usize) -> Self

pub fn with_enable_nnls(self, v: bool) -> Self

pub fn with_nnls_beta(self, v: f64) -> Self

pub fn with_calibration_lambda(self, v: f64) -> Self

pub fn with_batch_size(self, v: Option<usize>) -> Self

Trait Implementations

impl Default for EnsembleParams

fn default() -> Self

Returns the “default value” for a type.