com.feedzai.openml.provider.lightgbm

Class LightGBMTreeSHAPFeatureContributionExplainer

java.lang.Object

com.feedzai.openml.provider.lightgbm.LightGBMTreeSHAPFeatureContributionExplainer

All Implemented Interfaces:

com.feedzai.openml.explanations.ModelExplainer

public class LightGBMTreeSHAPFeatureContributionExplainer
extends Object
implements com.feedzai.openml.explanations.ModelExplainer

A TreeSHAP feature contribution provider for the predictions of the LightGBMBinaryClassificationModel.

Author:

Miguel Lobo (miguel.lobo@feedzai.com)

Since:

1.3.0

Constructor Summary

Constructors

Constructor and Description
LightGBMTreeSHAPFeatureContributionExplainer(LightGBMBinaryClassificationModel model) Constructor.

Method Summary

Modifier and Type	Method and Description
double[]	getFeatureContributions(com.feedzai.openml.data.Instance instance)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LightGBMTreeSHAPFeatureContributionExplainer

public LightGBMTreeSHAPFeatureContributionExplainer(LightGBMBinaryClassificationModel model)

Constructor.

Parameters:

model - The LightGBMBinaryClassificationModel whose predictions will be explained.

Method Detail

getFeatureContributions

public double[] getFeatureContributions(com.feedzai.openml.data.Instance instance)

Specified by:

getFeatureContributions in interface com.feedzai.openml.explanations.ModelExplainer