Package org.neo4j.gds.ml.decisiontree
Class GiniIndex
java.lang.Object
org.neo4j.gds.ml.decisiontree.GiniIndex
- All Implemented Interfaces:
ImpurityCriterion
-
Nested Class Summary
Nested classes/interfaces inherited from interface org.neo4j.gds.ml.decisiontree.ImpurityCriterion
ImpurityCriterion.ImpurityData -
Constructor Summary
ConstructorsConstructorDescriptionGiniIndex(org.neo4j.gds.collections.ha.HugeIntArray expectedMappedLabels, int numberOfClasses) -
Method Summary
Modifier and TypeMethodDescriptionvoiddecrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData) org.neo4j.gds.ml.decisiontree.GiniIndex.GiniImpurityDatagroupImpurity(org.neo4j.gds.collections.ha.HugeLongArray group, long startIndex, long size) voidincrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData) static org.neo4j.gds.core.utils.mem.MemoryRangememoryEstimation(long numberOfTrainingSamples) Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitMethods 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:
groupImpurityin interfaceImpurityCriterion
-
incrementalImpurity
- Specified by:
incrementalImpurityin interfaceImpurityCriterion
-
decrementalImpurity
- Specified by:
decrementalImpurityin interfaceImpurityCriterion
-