Enum forust_ml::gradientbooster::MissingNodeTreatment

pub enum MissingNodeTreatment {
    None,
    AssignToParent,
    AverageLeafWeight,
    AverageNodeWeight,
}

Variants

None

Calculate missing node weight values without any constraints.

AssignToParent

Assign the weight of the missing node to that of the parent.

AverageLeafWeight

After training each tree, starting from the bottom of the tree, assign the missing node weight to the weighted average of the left and right child nodes. Next assign the parent to the weighted average of the children nodes. This is performed recursively up through the entire tree. This is performed as a post processing step on each tree after it is built, and prior to updating the predictions for which to train the next tree.

AverageNodeWeight

Set the missing node to be equal to the weighted average weight of the left and the right nodes.

Trait Implementations

impl Clone for MissingNodeTreatment

fn clone(&self) -> MissingNodeTreatment

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<'de> Deserialize<'de> for MissingNodeTreatment

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for MissingNodeTreatment

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for MissingNodeTreatment