predict.aml_gbm: AML GBM predict method

Description Usage Arguments Value

View source: R/trees.R

Description

Returns prediction for a GBM when given a row of data.

Usage

1
2
## S3 method for class 'aml_gbm'
predict(gbm, data, n_trees = NULL)

Arguments

gbm

Ensemble of trees trained by 'aml_gbm'

data

Data.frame row of size '1 x p' for prediction

n_trees

Number of trees used in prediction, uses all trees by default

Value

Prediction value


jmwerner/ArtisanalMachineLearning documentation built on Jan. 7, 2020, 1:50 a.m.