Package org.neo4j.gds.ml.models.randomforest

Class RandomForestClassifier

java.lang.Object

org.neo4j.gds.ml.models.randomforest.RandomForestClassifier

All Implemented Interfaces:

Classifier

public class RandomForestClassifier
extends Object
implements Classifier

Nested Class Summary

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

Classifier.ClassifierData

Constructor Summary

Constructors

Constructor	Description
RandomForestClassifier(List<DecisionTreePredictor<Integer>> decisionTrees, int numberOfClasses, int featureDimension)
RandomForestClassifier(RandomForestClassifierData data)

Method Summary

Modifier and Type	Method	Description
Classifier.ClassifierData	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 numberOfClasses)

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

Constructor Details

RandomForestClassifier

public RandomForestClassifier(List<DecisionTreePredictor<Integer>> decisionTrees,
 int numberOfClasses,
 int featureDimension)

RandomForestClassifier

public RandomForestClassifier(RandomForestClassifierData data)

Method Details

runtimeOverheadMemoryEstimation

public static org.neo4j.gds.core.utils.mem.MemoryRange runtimeOverheadMemoryEstimation(int numberOfClasses)

data

public Classifier.ClassifierData data()

Specified by:

data in interface Classifier

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