Constructor Detail

• InformationGainFunction

  public InformationGainFunction(int numOfFeatures,
                                 Random random)

  Construct an information gain split function (it needs to be initialize with the corresponding data and indices)

  Parameters:

  numOfFeatures - number of random features to use

  random - random number generator

Method Detail

• init

  public void init(Instances data,
                   ArrayList<Integer> indices)

  Initialize the function given a specific data set

  Specified by:

  init in class SplitFunction

  Parameters:

  data - link to the original data

  indices - indices of the samples to use

• evaluate

  public boolean evaluate(Instance instance)

  Evaluate a single instance based on the current state of the split function

  Specified by:

  evaluate in class SplitFunction

  Parameters:

  instance - sample to evaluate

  Returns:

  false if the instance is on the right of the splitting point, true if it's on the left

• newInstance

  public SplitFunction newInstance()

  Specified by:

  newInstance in class SplitFunction