com.feedzai.openml.provider.lightgbm

Class LightGBMMLProvider

java.lang.Object

com.feedzai.openml.provider.lightgbm.LightGBMMLProvider

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 LightGBMMLProvider
extends Object
implements com.feedzai.openml.provider.TrainingMachineLearningProvider<LightGBMModelCreator>

This class implements Feedzai's OpenML MachineLearningProvider interface for LightGBM.

Author:

Alberto Ferreira (alberto.ferreira@feedzai.com)

Since:

1.0.10

Constructor Summary

Constructors

Constructor and Description
LightGBMMLProvider()

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

LightGBMMLProvider

public LightGBMMLProvider()

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>