ImpurityCriterion (ml-algo 2.7.0-alpha01 API)

All Known Implementing Classes:: Entropy, GiniIndex, SplitMeanSquaredError

public interface ImpurityCriterion

Nested Class Summary

Nested Classes

Modifier and Type

Interface

Description

static interface

ImpurityCriterion.ImpurityData

A lightweight representation of a decision tree node's impurity.
Method Summary

Modifier and Type

Method

Description

default double

combinedImpurity(ImpurityCriterion.ImpurityData leftImpurityData, ImpurityCriterion.ImpurityData rightImpurityData)

void

decrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData)

ImpurityCriterion.ImpurityData

groupImpurity(org.neo4j.gds.collections.ha.HugeLongArray group, long startIdx, long size)

void

incrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData)

Method Details
- groupImpurity
  
  ImpurityCriterion.ImpurityData groupImpurity(org.neo4j.gds.collections.ha.HugeLongArray group, long startIdx, long size)
- incrementalImpurity
  
  void incrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData)
- decrementalImpurity
  
  void decrementalImpurity(long featureVectorIdx, ImpurityCriterion.ImpurityData impurityData)
- combinedImpurity
  
  default double combinedImpurity(ImpurityCriterion.ImpurityData leftImpurityData, ImpurityCriterion.ImpurityData rightImpurityData)