Laurae (version 0.0.0.9001)

predictor_xgb: Partial Dependency, xgboost predictor

Description

This function is a helper for partial dependency plots using an xgboost model. Use this as an example for the predictor argument.

Usage

predictor_xgb(model, data)

Arguments

model

Type: unknown. The xgboost trained model.

data

Type: data.table (mandatory). The data we need to use to sample from for the partial dependency with observation.

Value

A vector of predicted values matching the right order of input.