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

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

import com.expleague.commons.func.Factory;
import com.expleague.ml.methods.VecOptimization;
import com.expleague.ml.methods.multiclass.gradfac.GradFacSvdNMulticlass;
import com.expleague.ml.data.tools.DataTools;
import com.expleague.ml.loss.L2;

/**
 * User: qdeee
 * Date: 03.09.14
 */
public class MultiClassSplitGradFacSvdNBuilder implements Factory {
  private final Factory defaultWeakBuilder = new BootstrapOptimizationBuilder();

  private VecOptimization weak;
  private String localName = "SatL2";
  private int factorDim = 1;
  private boolean needCompact = true;
  private boolean printErrors = false;
  private boolean bootstrap = false;

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

  public void setLocal(final String localName) {
    this.localName = localName;
  }

  public void setDim(final int factorDim) {
    this.factorDim = factorDim;
  }

  public void setCompact(final boolean needCompact) {
    this.needCompact = needCompact;
  }

  public void setOut(final boolean printErrors) {
    this.printErrors = printErrors;
  }

  public void setBootstrap(final boolean enable) {
    this.bootstrap = enable;
  }

  @Override
  public GradFacSvdNMulticlass create() {
    if (weak == null) {
      weak = defaultWeakBuilder.create();
    }
    return new GradFacSvdNMulticlass(weak, (Class) DataTools.targetByName(localName), factorDim, needCompact, printErrors);
  }}