com.feedzai.openml.provider.lightgbm

Class LightGBMBinaryClassificationModel

java.lang.Object

com.feedzai.openml.provider.lightgbm.LightGBMBinaryClassificationModel

All Implemented Interfaces:

com.feedzai.openml.model.ClassificationMLModel, com.feedzai.openml.model.MachineLearningModel, AutoCloseable

public class LightGBMBinaryClassificationModel
extends Object
implements com.feedzai.openml.model.ClassificationMLModel

This class is responsible for loading a saved LightGBM model binary and scoring instances.

Author:

Alberto Ferreira (alberto.ferreira@feedzai.com)

Since:

1.0.10

Method Summary

Modifier and Type	Method and Description
int	classify(com.feedzai.openml.data.Instance instance)
void	close()
String[]	getBoosterFeatureNames()
int	getBoosterNumFeatures()
int	getBoosterNumIterations()
double[]	getClassDistribution(com.feedzai.openml.data.Instance instance)
com.feedzai.openml.data.schema.DatasetSchema	getSchema()
boolean	isModelBinary()
boolean	save(Path dir, String name)

Methods inherited from class java.lang.Object

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

Method Detail

getClassDistribution

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

Specified by:

getClassDistribution in interface com.feedzai.openml.model.ClassificationMLModel

classify

public int classify(com.feedzai.openml.data.Instance instance)

Specified by:

classify in interface com.feedzai.openml.model.ClassificationMLModel

save

public boolean save(Path dir,
                    String name)

Specified by:

save in interface com.feedzai.openml.model.MachineLearningModel

getSchema

public com.feedzai.openml.data.schema.DatasetSchema getSchema()

Specified by:

getSchema in interface com.feedzai.openml.model.MachineLearningModel

close

public void close()
           throws Exception

Specified by:

close in interface AutoCloseable

Throws:

Exception

getBoosterNumFeatures

public int getBoosterNumFeatures()

Returns:

Number of features used in the model.

getBoosterFeatureNames

public String[] getBoosterFeatureNames()

Returns:

Names of the features in the model.

Since:

1.0.18

getBoosterNumIterations

public int getBoosterNumIterations()

Returns:

The number of booster iterations in the model.

isModelBinary

public boolean isModelBinary()

Returns:

Returns true if the model is binary.