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

Class LogisticRegressionClassifier

java.lang.Object

org.neo4j.gds.ml.models.logisticregression.LogisticRegressionClassifier

All Implemented Interfaces:

Classifier

public final class LogisticRegressionClassifier
extends Object
implements Classifier

Nested Class Summary

Nested classes/interfaces inherited from interface org.neo4j.gds.ml.models.Classifier

Classifier.ClassifierData

Method Summary

Modifier and Type	Method	Description
LogisticRegressionData	data()
static LogisticRegressionClassifier	from(LogisticRegressionData data)
double[]	predictProbabilities(double[] features)
org.neo4j.gds.ml.core.tensor.Matrix	predictProbabilities(org.neo4j.gds.ml.core.batch.Batch batch, Features features)
static org.neo4j.gds.core.utils.mem.MemoryRange	runtimeOverheadMemoryEstimation(int batchSize, int featureDimension, int numberOfClasses, boolean isReduced)
static long	sizeOfPredictionsVariableInBytes(int batchSize, int numberOfFeatures, int numberOfClasses, int normalizedNumberOfClasses)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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

numberOfClasses

Method Details

from

public static LogisticRegressionClassifier from(LogisticRegressionData data)

sizeOfPredictionsVariableInBytes

public static long sizeOfPredictionsVariableInBytes(int batchSize,
 int numberOfFeatures,
 int numberOfClasses,
 int normalizedNumberOfClasses)

runtimeOverheadMemoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryRange runtimeOverheadMemoryEstimation(int batchSize,
 int featureDimension,
 int numberOfClasses,
 boolean isReduced)

predictProbabilities

public double[] predictProbabilities(double[] features)

Specified by:

predictProbabilities in interface Classifier

predictProbabilities

public org.neo4j.gds.ml.core.tensor.Matrix predictProbabilities(org.neo4j.gds.ml.core.batch.Batch batch,
 Features features)

Specified by:

predictProbabilities in interface Classifier

data

public LogisticRegressionData data()

Specified by:

data in interface Classifier