Skip to main content

TrainConfig

Struct TrainConfig 

Source
pub struct TrainConfig {
Show 16 fields pub algorithm: TrainAlgorithm, pub task: Task, pub tree_type: TreeType, pub criterion: Criterion, pub max_depth: Option<usize>, pub min_samples_split: Option<usize>, pub min_samples_leaf: Option<usize>, pub physical_cores: Option<usize>, pub n_trees: Option<usize>, pub max_features: MaxFeatures, pub seed: Option<u64>, pub compute_oob: bool, pub learning_rate: Option<f64>, pub bootstrap: bool, pub top_gradient_fraction: Option<f64>, pub other_gradient_fraction: Option<f64>,
}
Expand description

Unified training configuration shared by the Rust and Python entry points.

The crate keeps one normalized config type so the binding layer only has to perform input validation and type conversion; all semantic decisions happen from this one structure downward.

Fields§

§algorithm: TrainAlgorithm

High-level training family.

§task: Task

Regression or classification.

§tree_type: TreeType

Tree learner used by the selected algorithm family.

§criterion: Criterion

Split criterion. Criterion::Auto is resolved by the trainer.

§max_depth: Option<usize>

Maximum tree depth.

§min_samples_split: Option<usize>

Smallest node size that is still allowed to split.

§min_samples_leaf: Option<usize>

Minimum child size after a split.

§physical_cores: Option<usize>

Optional cap on training-side rayon threads.

§n_trees: Option<usize>

Number of trees for ensemble algorithms.

§max_features: MaxFeatures

Feature subsampling strategy.

§seed: Option<u64>

Seed used for reproducible sampling and randomized splits.

§compute_oob: bool

Whether random forests should compute out-of-bag metrics.

§learning_rate: Option<f64>

Gradient boosting shrinkage factor.

§bootstrap: bool

Whether gradient boosting should bootstrap rows before gradient sampling.

§top_gradient_fraction: Option<f64>

Fraction of largest-gradient rows always kept by GOSS sampling.

§other_gradient_fraction: Option<f64>

Fraction of the remaining rows randomly retained by GOSS sampling.

Trait Implementations§

Source§

impl Clone for TrainConfig

Source§

fn clone(&self) -> TrainConfig

Returns a duplicate of the value. Read more
1.0.0 · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Debug for TrainConfig

Source§

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

Formats the value using the given formatter. Read more
Source§

impl Default for TrainConfig

Source§

fn default() -> Self

Returns the “default value” for a type. Read more
Source§

impl Copy for TrainConfig

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> DynClone for T
where T: Clone,

Source§

fn __clone_box(&self, _: Private) -> *mut ()

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Pointable for T

Source§

const ALIGN: usize

The alignment of pointer.
Source§

type Init = T

The type for initializers.
Source§

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more
Source§

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more
Source§

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more
Source§

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<T> Allocation for T
where T: RefUnwindSafe + Send + Sync,