Package org.neo4j.gds.ml.decisiontree

Class Entropy

java.lang.Object

org.neo4j.gds.ml.decisiontree.Entropy

All Implemented Interfaces:

ImpurityCriterion

public class Entropy
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
Entropy(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.Entropy.EntropyImpurityData	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

Entropy

public Entropy(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.Entropy.EntropyImpurityData 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