com.feedzai.openml.provider.lightgbm

Class FairGBMMLProvider

java.lang.Object

com.feedzai.openml.provider.lightgbm.FairGBMMLProvider

All Implemented Interfaces:

com.feedzai.openml.provider.MachineLearningProvider<LightGBMModelCreator>, com.feedzai.openml.provider.TrainingMachineLearningProvider<LightGBMModelCreator>

@AutoService(value=com.feedzai.openml.provider.MachineLearningProvider.class)
public class FairGBMMLProvider
extends Object
implements com.feedzai.openml.provider.TrainingMachineLearningProvider<LightGBMModelCreator>

This class implements Feedzai's OpenML MachineLearningProvider interface for FairGBM (constrained LightGBM).

Author:

Andre Cruz (andre.cruz@feedzai.com)

Since:

1.4.0

Constructor Summary

Constructors

Constructor and Description
FairGBMMLProvider()

Method Summary

Modifier and Type	Method and Description
Set<com.feedzai.openml.provider.descriptor.MLAlgorithmDescriptor>	getAlgorithms()
Optional<LightGBMModelCreator>	getModelCreator(String algorithmName)
String	getName()

Methods inherited from class java.lang.Object

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

Constructor Detail

FairGBMMLProvider

public FairGBMMLProvider()

Method Detail

getName

public String getName()

Specified by:

getName in interface com.feedzai.openml.provider.MachineLearningProvider<LightGBMModelCreator>

getAlgorithms

public Set<com.feedzai.openml.provider.descriptor.MLAlgorithmDescriptor> getAlgorithms()

Specified by:

getAlgorithms in interface com.feedzai.openml.provider.MachineLearningProvider<LightGBMModelCreator>

getModelCreator

public Optional<LightGBMModelCreator> getModelCreator(String algorithmName)

Specified by:

getModelCreator in interface com.feedzai.openml.provider.MachineLearningProvider<LightGBMModelCreator>