gbm_fit: Gradient boosting machine based on GPR

Description Usage Arguments Value

View source: R/boosting_gpr.R

Description

GBM prediction

Usage

1
gbm_fit(testmx, test_y, trainmx, gbm_model, ncpu = -1)

Arguments

testmx

Matrix; the features of testing dateset.

trainmx

Matrix; the features of training data set.

gbm_model

List; the output of gbm_train(), containing three objects: models pred_method train_rmse

ncpu

Integer; the number of thread to be used; set to -1 to use all threads; default value is -1.

Value

List; prediction: the final prediction on testing dataset. test_rmse: numeric vector; list of rmse during testing.


weichunliao/baeirGPR documentation built on July 30, 2018, 10:06 a.m.