Gradient boosting decision tree

Hierarchy (View Summary)

Constructors

Properties

Accessors

Methods

Constructors

  • Parameters

    • Optionalmaxdepth: number

      Maximum depth of tree

    • Optionalsrate: number

      Sampling rate

    • Optionallr: number

      Learning rate

    Returns GBDT

Properties

_loss: Matrix<number[]>
_lr: number
_maxd: number
_r: any[]
_srate: number
_trees: any[]
_x: number[][]
_y: Matrix<number[]>

Accessors

  • get size(): number
  • Number of trees

    Returns number

Methods

  • Parameters

    • n: any

    Returns number[]

  • Fit model.

    Returns void

  • Initialize model.

    Parameters

    • x: number[][]

      Training data

    • y: number[][]

      Target values

    Returns void

  • Returns predicted values.

    Parameters

    • x: number[][]

      Sample data

    Returns number[][]

    Predicted values