com.expleague.ml.cli.builders.methods.impl.MultiClassSplitGradFacSvdNBuilder 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.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);
  }}