Package org.neo4j.gds.ml.gradientdescent

Interface GradientDescentConfig

All Superinterfaces:

org.neo4j.gds.config.ToleranceConfig, org.neo4j.gds.config.ToMapConvertible

All Known Subinterfaces:

LinearRegressionTrainConfig, LogisticRegressionTrainConfig, MLPClassifierTrainConfig

@Configuration
public interface GradientDescentConfig
extends org.neo4j.gds.config.ToleranceConfig, org.neo4j.gds.config.ToMapConvertible

Field Summary

Fields

Modifier and Type	Field	Description
static final int	DEFAULT_BATCH_SIZE
static final int	MAX_EPOCHS

Method Summary

Modifier and Type	Method	Description
default int	batchSize()
default double	learningRate()
default int	maxEpochs()
default int	minEpochs()
default int	patience()
default double	tolerance()
Map<String,Object>	toMap()

Field Details

DEFAULT_BATCH_SIZE

static final int DEFAULT_BATCH_SIZE

See Also:

• Constant Field Values

MAX_EPOCHS

static final int MAX_EPOCHS

See Also:

• Constant Field Values

Method Details

batchSize

@IntegerRange(min=1)
default int batchSize()

minEpochs

@IntegerRange(min=1)
default int minEpochs()

patience

@IntegerRange(min=1)
default int patience()

maxEpochs

@IntegerRange(min=1)
default int maxEpochs()

tolerance

@DoubleRange(min=0.0)
default double tolerance()

Specified by:

tolerance in interface org.neo4j.gds.config.ToleranceConfig

learningRate

default double learningRate()

toMap

@ToMap
Map<String,Object> toMap()

Specified by:

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