Class XGBoost

eXtreme Gradient Boosting regression

Hierarchy (View Summary)

XGBoost
- XGBoostClassifier

Index

Constructors

Properties

_lambda _learning_rate _loss _maxd _r _srate _trees _x _y

Accessors

Methods

_sample fit init predict

Constructors

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
- Defined in lib/model/xgboost.d.ts:11

Properties

_lambda

_lambda: number

_learning_rate

_learning_rate: number

_loss

_loss: Matrix<number[]>

_maxd

_maxd: number

_r

_r: any[]

_srate

_srate: number

_trees

_trees: any[]

_x

_x: number[][]

_y

_y: Matrix<number[]>

Accessors

size

get size(): number
Number of trees

Returns number
- Defined in lib/model/xgboost.d.ts:22

Methods

_sample

_sample(n: any): number[]
Parameters
- n: any
Returns number[]
- Defined in lib/model/xgboost.d.ts:23

fit

fit(): void
Fit model.

Returns void
- Defined in lib/model/xgboost.d.ts:36

init

init(x: number[][], y: number[][]): void
Initialize model.
Parameters
- x: number[][]
  Training data
- y: number[][]
  Target values
Returns void
- Defined in lib/model/xgboost.d.ts:29

predict

predict(x: number[][]): number[][]
Returns predicted values.
Parameters
- x: number[][]
  Sample data
Returns number[][]
Predicted values
- Defined in lib/model/xgboost.d.ts:42