Package org.neo4j.gds.ml.decisiontree

Class SplitMeanSquaredError

java.lang.Object

org.neo4j.gds.ml.decisiontree.SplitMeanSquaredError

All Implemented Interfaces:

ImpurityCriterion

public class SplitMeanSquaredError
extends Object
implements ImpurityCriterion

Nested Class Summary

Nested classes/interfaces inherited from interface org.neo4j.gds.ml.decisiontree.ImpurityCriterion

ImpurityCriterion.ImpurityData

Constructor Summary

Constructors

Constructor	Description
SplitMeanSquaredError(org.neo4j.gds.collections.ha.HugeDoubleArray targets)

Method Summary

Modifier and Type	Method	Description
void	decrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData)
org.neo4j.gds.ml.decisiontree.SplitMeanSquaredError.MSEImpurityData	groupImpurity(org.neo4j.gds.collections.ha.HugeLongArray group, long startIdx, long size)
void	incrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData)
static org.neo4j.gds.core.utils.mem.MemoryRange	memoryEstimation()

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.decisiontree.ImpurityCriterion

combinedImpurity

Constructor Details

SplitMeanSquaredError

public SplitMeanSquaredError(org.neo4j.gds.collections.ha.HugeDoubleArray targets)

Method Details

memoryEstimation

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

groupImpurity

public org.neo4j.gds.ml.decisiontree.SplitMeanSquaredError.MSEImpurityData groupImpurity(org.neo4j.gds.collections.ha.HugeLongArray group,
 long startIdx,
 long size)

Specified by:

groupImpurity in interface ImpurityCriterion

incrementalImpurity

public void incrementalImpurity(long featureVectorIdx,
 ImpurityCriterion.ImpurityData impurityData)

Specified by:

incrementalImpurity in interface ImpurityCriterion

decrementalImpurity

public void decrementalImpurity(long featureVectorIdx,
 ImpurityCriterion.ImpurityData impurityData)

Specified by:

decrementalImpurity in interface ImpurityCriterion