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

Go to download
package com.expleague.ml.cli.builders.methods.impl;

import com.expleague.commons.random.FastRandom;
import com.expleague.ml.BFGrid;
import com.expleague.ml.methods.VecOptimization;
import com.expleague.ml.methods.greedyRegion.RegionForest;
import com.expleague.commons.func.Factory;

/**
 * User: noxoomo
 * Date: 10.11.14
 */

public class RegionForestBuilder implements Factory {
  public static Factory defaultGridBuilder;
  private double alpha = 0.02;
  private double beta = 0.5;
  private int maxFailed = 1;
  private int regionsCount = 5;
  private String meanMethod = "Naive";

  private Factory gridBuilder = defaultGridBuilder;

  public void setGridBuilder(final Factory gridBuilder) {
    this.gridBuilder = gridBuilder;
  }

  public void setAlpha(final double alpha) {
    this.alpha = alpha;
  }

  public void setBeta(final double beta) {
    this.beta = beta;
  }

  public void setMaxFailed(final int maxFailed) {
    this.maxFailed = maxFailed;
  }

  public void setRegionsCount(final int regionsCount) {
    this.regionsCount = regionsCount;
  }

  public void setMeanMethod(final String method) {
    this.meanMethod = method;
  }


  @Override
  public VecOptimization create() {
    return new RegionForest(gridBuilder.create(), new FastRandom(), regionsCount, RegionForest.MeanMethod.valueOf(meanMethod), alpha, beta, maxFailed);
  }
}