gbm_fit: Gradient boosting machine based on GPR

In weichunliao/baeirGPR: Traing a Gaussian Process Regression and Conduct Predictions.

Description

GBM prediction

Usage

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 May 18, 2019, 9:15 p.m.