Package org.neo4j.gds.ml.decisiontree

Class GiniIndex

java.lang.Object

org.neo4j.gds.ml.decisiontree.GiniIndex

All Implemented Interfaces:

ImpurityCriterion

public class GiniIndex
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
GiniIndex(org.neo4j.gds.collections.ha.HugeIntArray expectedMappedLabels, int numberOfClasses)

Method Summary

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

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

GiniIndex

public GiniIndex(org.neo4j.gds.collections.ha.HugeIntArray expectedMappedLabels,
 int numberOfClasses)

Method Details

memoryEstimation

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

groupImpurity

public org.neo4j.gds.ml.decisiontree.GiniIndex.GiniImpurityData groupImpurity(org.neo4j.gds.collections.ha.HugeLongArray group,
 long startIndex,
 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