Struct LearnerConfig

pub struct LearnerConfig {
    pub num_classes: u32,
    pub regularization: Option<TreeRegularizationConfig>,
    pub constraints: Option<TreeConstraintsConfig>,
    pub pruning_mode: i32,
    pub growing_mode: i32,
    pub learning_rate_tuner: Option<LearningRateConfig>,
    pub multi_class_strategy: i32,
    pub averaging_config: Option<AveragingConfig>,
    pub weak_learner_type: i32,
    pub each_tree_start: Option<DecisionTreeConfig>,
    pub each_tree_start_num_layers: i32,
    pub feature_fraction: Option<FeatureFraction>,
}

Fields

num_classes: u32

Number of classes.

regularization: Option<TreeRegularizationConfig>

Regularization.

constraints: Option<TreeConstraintsConfig>

Constraints.

pruning_mode: i32

Pruning. POST_PRUNE is the default pruning mode.

growing_mode: i32

Growing Mode. LAYER_BY_LAYER is the default growing mode.

learning_rate_tuner: Option<LearningRateConfig>

Learning rate. By default we use fixed learning rate of 0.1.

multi_class_strategy: i32

Multi-class strategy. By default we use TREE_PER_CLASS for binary classification and linear regression. For other cases, we use DIAGONAL_HESSIAN as the default.

averaging_config: Option<AveragingConfig>

If you want to average the ensembles (for regularization), provide the config below.

weak_learner_type: i32

By default we use NORMAL_DECISION_TREE as weak learner.

each_tree_start: Option<DecisionTreeConfig>

If you want to enforce some splits and allow boosting to figure out the rest, you can provide a tree that represents the starting splits for each tree in the ensemble. Set both each_tree_start and each_tree_start_num_layers.

each_tree_start_num_layers: i32

feature_fraction: Option<FeatureFraction>

Fraction of features to consider in each tree sampled randomly from all available features.

Implementations

impl LearnerConfig

pub fn pruning_mode(&self) -> PruningMode

Returns the enum value of pruning_mode, or the default if the field is set to an invalid enum value.

pub fn set_pruning_mode(&mut self, value: PruningMode)

Sets pruning_mode to the provided enum value.

pub fn growing_mode(&self) -> GrowingMode

Returns the enum value of growing_mode, or the default if the field is set to an invalid enum value.

pub fn set_growing_mode(&mut self, value: GrowingMode)

Sets growing_mode to the provided enum value.

pub fn multi_class_strategy(&self) -> MultiClassStrategy

Returns the enum value of multi_class_strategy, or the default if the field is set to an invalid enum value.

pub fn set_multi_class_strategy(&mut self, value: MultiClassStrategy)

Sets multi_class_strategy to the provided enum value.

pub fn weak_learner_type(&self) -> WeakLearnerType

Returns the enum value of weak_learner_type, or the default if the field is set to an invalid enum value.

pub fn set_weak_learner_type(&mut self, value: WeakLearnerType)

Sets weak_learner_type to the provided enum value.

Trait Implementations

impl Clone for LearnerConfig

fn clone(&self) -> LearnerConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LearnerConfig

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

Formats the value using the given formatter.

impl Default for LearnerConfig

fn default() -> Self

Returns the “default value” for a type.

impl Message for LearnerConfig

fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

fn clear(&mut self)

Clears the message, resetting all fields to their default.

fn encode<B>(&self, buf: &mut B) -> Result<(), EncodeError>

where B: BufMut, Self: Sized,

Encodes the message to a buffer.

fn encode_length_delimited<B>(&self, buf: &mut B) -> Result<(), EncodeError>

where B: BufMut, Self: Sized,

Encodes the message with a length-delimiter to a buffer.

fn decode<B>(buf: B) -> Result<Self, DecodeError>

where B: Buf, Self: Default,

Decodes an instance of the message from a buffer.

fn decode_length_delimited<B>(buf: B) -> Result<Self, DecodeError>

where B: Buf, Self: Default,

Decodes a length-delimited instance of the message from the buffer.

fn merge<B>(&mut self, buf: B) -> Result<(), DecodeError>

where B: Buf, Self: Sized,

Decodes an instance of the message from a buffer, and merges it into self.

fn merge_length_delimited<B>(&mut self, buf: B) -> Result<(), DecodeError>

where B: Buf, Self: Sized,

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq for LearnerConfig

fn eq(&self, other: &LearnerConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for LearnerConfig