org.apache.ignite.ml.trainers

Class MultiLabelDatasetTrainer<M extends IgniteModel>

java.lang.Object

org.apache.ignite.ml.trainers.DatasetTrainer<M,double[]>

org.apache.ignite.ml.trainers.MultiLabelDatasetTrainer<M>

Type Parameters:

M - Type of a produced model.

Direct Known Subclasses:

MLPTrainer

public abstract class MultiLabelDatasetTrainer<M extends IgniteModel>
extends DatasetTrainer<M,double[]>

Interface for trainers that trains on dataset with multiple label per object.

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

DatasetTrainer.EmptyDatasetException

Field Summary

Fields inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

envBuilder, environment

Constructor Summary

Constructors

Constructor and Description
MultiLabelDatasetTrainer()

Method Summary

Methods inherited from class org.apache.ignite.ml.trainers.DatasetTrainer

fit, fit, fit, fit, fit, fit, fitWithInitializedDeployingContext, getLastTrainedModelOrThrowEmptyDatasetException, identityTrainer, isUpdateable, learningEnvironment, update, update, update, update, update, updateModel, withConvertedLabels, withEnvironmentBuilder

Methods inherited from class java.lang.Object

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

Constructor Detail

MultiLabelDatasetTrainer

public MultiLabelDatasetTrainer()