com.expleague.ml.cli.builders.methods.impl.RandomForestBuilder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jmll

Various ML methods implemented by myself and my students

There is a newer version: 1.4.9

package com.expleague.ml.cli.builders.methods.impl;

import com.expleague.ml.loss.WeightedLoss;
import com.expleague.ml.methods.VecOptimization;
import com.expleague.commons.func.Factory;
import com.expleague.commons.random.FastRandom;
import com.expleague.ml.methods.RandomForest;

/**
 * User: qdeee
 * Date: 27.01.15
 */
public class RandomForestBuilder implements Factory {
  public static FastRandom defaultRandom;

  private final Factory defaultWeakBuilder = new GreedyObliviousTreeBuilder();
  private FastRandom random = defaultRandom;
  private int count;
  private VecOptimization weak;

  public void setCount(final int count) {
    this.count = count;
  }

  public void setWeak(final VecOptimization weak) {
    this.weak = weak;
  }

  @Override
  public RandomForest create() {
    if (weak == null) {
      weak = defaultWeakBuilder.create();
    }

    return new RandomForest(weak, random, count);
  }
}