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

Interface MLPClassifierData

All Superinterfaces:
BaseModelData, Classifier.ClassifierData, Serializable
public interface MLPClassifierData extends Classifier.ClassifierData, Serializable

  • Method Summary

    Modifier and Type
    Method
    Description
    List<org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Vector>>
    biases()
     
    static org.neo4j.gds.ml.models.mlp.ImmutableMLPClassifierData.Builder
    builder()
     
    static MLPClassifierData
    create(int classCount, int featureCount, List<Integer> hiddenLayerSizes, SplittableRandom random)
     
    default int
    depth()
     
    default int
    featureDimension()
     
    default int
    numberOfClasses()
     
    default org.neo4j.gds.ml.api.TrainingMethod
    trainerMethod()
     
    List<org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix>>
    weights()
     

  • Method Details

    • weights

      List<org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix>> weights()

    • biases

      List<org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Vector>> biases()

    • depth

      @Derived default int depth()

    • numberOfClasses

      @Derived default int numberOfClasses()
      Specified by:
      numberOfClasses in interface Classifier.ClassifierData

    • featureDimension

      @Derived default int featureDimension()
      Specified by:
      featureDimension in interface BaseModelData

    • trainerMethod

      default org.neo4j.gds.ml.api.TrainingMethod trainerMethod()
      Specified by:
      trainerMethod in interface BaseModelData

    • create

      static MLPClassifierData create(int classCount, int featureCount, List<Integer> hiddenLayerSizes, SplittableRandom random)

    • builder

      static org.neo4j.gds.ml.models.mlp.ImmutableMLPClassifierData.Builder builder()