Package org.neo4j.gds.ml.models.linearregression

Interface LinearRegressionData

All Superinterfaces:
BaseModelData, Regressor.RegressorData
public interface LinearRegressionData extends Regressor.RegressorData

  • Method Summary

    Modifier and Type
    Method
    Description
    org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Scalar>
    bias()
     
    default int
    featureDimension()
     
    static LinearRegressionData
    of(int featureDimension)
     
    default org.neo4j.gds.ml.api.TrainingMethod
    trainerMethod()
     
    org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix>
    weights()
     

  • Method Details

    • weights

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

    • bias

      org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Scalar> bias()

    • trainerMethod

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

    • featureDimension

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

    • of

      static LinearRegressionData of(int featureDimension)