Package org.neo4j.gds.ml.models

Interface ClassAwareTrainerConfig

All Superinterfaces:

org.neo4j.gds.config.ToMapConvertible, TrainerConfig

All Known Subinterfaces:

LogisticRegressionTrainConfig, MLPClassifierTrainConfig

public interface ClassAwareTrainerConfig
extends TrainerConfig

Method Summary

Modifier and Type	Method	Description
default List<Double>	classWeights()
default double	focusWeight()
default double[]	initializeClassWeights(int numberOfClasses)

Methods inherited from interface org.neo4j.gds.config.ToMapConvertible

toMap

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

method, toMapWithTrainerMethod, toTunableConfig

Method Details

focusWeight

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

classWeights

@Default
default List<Double> classWeights()

initializeClassWeights

@Ignore
default double[] initializeClassWeights(int numberOfClasses)