Class XGBoost

constructor

• new XGBoost(
      maxdepth?: number,
      srate?: number,
      lambda?: number,
      lr?: number,
  ): XGBoost

• Parameters

  • Optionalmaxdepth: number

    Maximum depth of tree
  • Optionalsrate: number

    Sampling rate
  • Optionallambda: number

    Regularization parameter
  • Optionallr: number

    Learning rate

  Returns XGBoost

_lambda

_learning_rate

_loss

_maxd

_r

_srate

_trees

_x

_y

size

• get size(): number

• Number of trees

  Returns number

_sample

• _sample(n: any): number[]

• Parameters

  • n: any

  Returns number[]

fit

• fit(): void

• Fit model.

  Returns void

init

• init(x: number[][], y: number[][]): void

• Initialize model.

  Parameters

  • x: number[][]

    Training data
  • y: number[][]

    Target values

  Returns void

predict

• predict(x: number[][]): number[][]

• Returns predicted values.

  Parameters

  • x: number[][]

    Sample data

  Returns number[][]

  Predicted values