Table of Contents

Mojo struct

BDecisionTree

@memory_only
struct BDecisionTree

Aliases

  • __del__is_trivial = False
  • __moveinit__is_trivial = True
  • __copyinit__is_trivial = True

Fields

  • min_samples_split (Int)
  • max_depth (Int)
  • reg_lambda (Float32)
  • reg_alpha (Float32)
  • gamma (Float32)
  • n_bins (Int)
  • root (UnsafePointer[Node, MutAnyOrigin])

Implemented traits

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDestructible, Movable

Methods

__init__

fn __init__(out self, min_samples_split: Int = 10, max_depth: Int = 3, reg_lambda: Float32 = 1, reg_alpha: Float32 = 0, gamma: Float32 = 0, n_bins: Int = 0)

Args:

  • min_samples_split (Int)
  • max_depth (Int)
  • reg_lambda (Float32)
  • reg_alpha (Float32)
  • gamma (Float32)
  • n_bins (Int)
  • self (Self)

Returns:

Self

__del__

fn __del__(deinit self)

Args:

  • self (Self)

fit

fn fit(mut self, X: Matrix, g: Matrix, h: Matrix)

Args:

  • self (Self)
  • X (Matrix)
  • g (Matrix)
  • h (Matrix)

Raises:

predict

fn predict(self, X: Matrix) -> Matrix

Args:

  • self (Self)
  • X (Matrix)

Returns:

Matrix

Raises:

Edit this page