com.expleague.ml.methods.MultipleVecOptimization Maven / Gradle / Ivy

Go to download

package com.expleague.ml.methods;

import com.expleague.ml.data.set.VecDataSet;
import com.expleague.ml.TargetFunc;
import com.expleague.commons.math.Trans;

import java.lang.reflect.Array;

/**
 * User: noxoomo
 */

public interface MultipleVecOptimization  extends VecOptimization {
  Trans[] fit(VecDataSet[] learn, Loss[] loss);

  abstract class Stub implements MultipleVecOptimization {
    @Override
    public Trans fit(VecDataSet learn, Loss loss) {
      VecDataSet[] ds = new VecDataSet[1];
      Loss[] losses = (Loss[]) Array.newInstance(loss.getClass(),1);
      ds[0] = learn;
      losses[0] = loss;
      return fit(ds, losses)[0];
    }
  }
}