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

Go to download
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);
  }
}