com.expleague.ml.Binarize 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;

import com.expleague.ml.data.impl.BinarizedDataSet;
import com.expleague.ml.data.set.VecDataSet;
import com.expleague.ml.dynamicGrid.impl.BinarizedDynamicDataSet;
import com.expleague.ml.dynamicGrid.interfaces.DynamicGrid;

import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;

/**
 * User: solar
 * Date: 12.11.13
 * Time: 18:43
 */
public class Binarize implements Function {
  private final Map grids = new HashMap<>(1);
  private final Map dynamicGrids = new HashMap<>(1);
  private VecDataSet set;

  public synchronized BinarizedDataSet binarize(final BFGrid grid) {
    return grids.compute(grid, (key, value) -> value != null ? value : new BinarizedDataSet(set, grid));
  }

  public synchronized BinarizedDynamicDataSet binarize(final DynamicGrid grid) {
    return dynamicGrids.compute(grid, (key, value) -> value != null ? value : new BinarizedDynamicDataSet(set, grid));
  }

  @Override
  public Binarize apply(final VecDataSet argument) {
    set = argument;
    return this;
  }
}