plotImportanceMatrix: Plot the Importance Matrix for a Gradient Boosted Model

In PriceLab/FPML: Footprint Machine Learning

Description

Plot the importance matrix, showing the features with the largest gain, for a given gradient boosted model and the full path to a desired .PNG file containing the figure

Usage

plotImportanceMatrix(gbModel, X_train, filePath)

Arguments

gbModel	A gradient boosted model, generally obtained using the buildBoostedModel function
X_train	The matrix of training data used to created the supplied model. This is generally created as an element of the list returned by the prepModelData function
filePath	The path to the .PNG file that will be created by the function. This file will house the figure of the importance matrix

Value

A .PNG file containing the importance matrix figure and located at the supplied file path

PriceLab/FPML documentation built on May 28, 2019, 2:25 p.m.