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

Class RandomForestRegressor

java.lang.Object

org.neo4j.gds.ml.models.randomforest.RandomForestRegressor

All Implemented Interfaces:

Regressor

public class RandomForestRegressor
extends Object
implements Regressor

Nested Class Summary

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

Regressor.RegressorData

Constructor Summary

Constructors

Constructor	Description
RandomForestRegressor(List<DecisionTreePredictor<Double>> decisionTrees, int featureDimension)
RandomForestRegressor(RandomForestRegressorData data)

Method Summary

Modifier and Type	Method	Description
Regressor.RegressorData	data()
double	predict(double[] features)
static org.neo4j.gds.core.utils.mem.MemoryRange	runtimeOverheadMemoryEstimation()

Methods inherited from class java.lang.Object

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

Constructor Details

RandomForestRegressor

public RandomForestRegressor(List<DecisionTreePredictor<Double>> decisionTrees,
 int featureDimension)

RandomForestRegressor

public RandomForestRegressor(RandomForestRegressorData data)

Method Details

runtimeOverheadMemoryEstimation

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

data

public Regressor.RegressorData data()

Specified by:

data in interface Regressor

predict

public double predict(double[] features)

Specified by:

predict in interface Regressor