com.expleague.ml.cli.builders.methods.grid.DynamicGridBuilder Maven / Gradle / Ivy

Go to download

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

import com.expleague.ml.data.set.VecDataSet;
import com.expleague.ml.dynamicGrid.impl.BFDynamicGrid;
import com.expleague.commons.func.Factory;
import com.expleague.ml.dynamicGrid.interfaces.DynamicGrid;

/**
 * User: qdeee
 * Date: 03.09.14
 */
public class DynamicGridBuilder implements Factory {
  DynamicGrid cooked;
  int binsCount = 32;
  VecDataSet dataSet;

  public void setGrid(final DynamicGrid cooked) {
    this.cooked = cooked;
  }

  public void setBinsCount(final int binsCount) {
    this.binsCount = binsCount;
  }

  public void setDataSet(final VecDataSet dataSet) {
    this.dataSet = dataSet;
  }

  @Override
  public DynamicGrid create() {
    if (cooked == null) {
      cooked = new BFDynamicGrid(dataSet, binsCount);
    }
    return cooked;
  }
}