Package org.neo4j.gds.ml.models.mlp

Interface MLPClassifierTrainConfig

All Superinterfaces:

ClassAwareTrainerConfig, GradientDescentConfig, PenaltyConfig, org.neo4j.gds.config.ToleranceConfig, org.neo4j.gds.config.ToMapConvertible, TrainerConfig

@Configuration
public interface MLPClassifierTrainConfig
extends GradientDescentConfig, PenaltyConfig, ClassAwareTrainerConfig

Field Summary

Fields

Modifier and Type	Field	Description
static final MLPClassifierTrainConfig	DEFAULT

Fields inherited from interface org.neo4j.gds.ml.gradientdescent.GradientDescentConfig

DEFAULT_BATCH_SIZE, MAX_EPOCHS

Method Summary

Modifier and Type	Method	Description
Collection<String>	configKeys()
default List<Integer>	hiddenLayerSizes()
default org.neo4j.gds.ml.api.TrainingMethod	method()
static MLPClassifierTrainConfig	of(Map<String,Object> params)
Map<String,Object>	toMap()

Methods inherited from interface org.neo4j.gds.ml.models.ClassAwareTrainerConfig

classWeights, focusWeight, initializeClassWeights

Methods inherited from interface org.neo4j.gds.ml.gradientdescent.GradientDescentConfig

batchSize, learningRate, maxEpochs, minEpochs, patience, tolerance

Methods inherited from interface org.neo4j.gds.ml.models.PenaltyConfig

penalty

Methods inherited from interface org.neo4j.gds.ml.models.TrainerConfig

toMapWithTrainerMethod, toTunableConfig

Field Details

DEFAULT

static final MLPClassifierTrainConfig DEFAULT

Method Details

hiddenLayerSizes

default List<Integer> hiddenLayerSizes()

toMap

@ToMap
Map<String,Object> toMap()

Specified by:

toMap in interface GradientDescentConfig

Specified by:

toMap in interface org.neo4j.gds.config.ToMapConvertible

configKeys

@CollectKeys
Collection<String> configKeys()

of

static MLPClassifierTrainConfig of(Map<String,Object> params)

method

@Ignore
default org.neo4j.gds.ml.api.TrainingMethod method()

Specified by:

method in interface TrainerConfig