Package org.neo4j.gds.ml.models.logisticregression

Interface LogisticRegressionTrainConfig

All Superinterfaces:

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

@Configuration
public interface LogisticRegressionTrainConfig
extends GradientDescentConfig, PenaltyConfig, ClassAwareTrainerConfig

Field Summary

Fields

Modifier and Type	Field	Description
static final LogisticRegressionTrainConfig	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 org.neo4j.gds.ml.api.TrainingMethod	method()
static LogisticRegressionTrainConfig	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 LogisticRegressionTrainConfig DEFAULT

Method Details

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 LogisticRegressionTrainConfig of(Map<String,Object> params)

method

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

Specified by:

method in interface TrainerConfig