com.expleague.ml.methods.rvm.RVM 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.methods.rvm;

import com.expleague.commons.random.FastRandom;
import com.expleague.ml.data.set.VecDataSet;
import com.expleague.ml.func.BiasedLinear;
import com.expleague.ml.methods.VecOptimization;
import com.expleague.ml.loss.L2;

/**
 * Created by noxoomo on 03/06/15.
 */
public class RVM extends VecOptimization.Stub {
  private double tolerance = 1e-3;
  private FastRandom random;

  public RVM(FastRandom random) {
    this.random = new FastRandom(random.nextLong());
  }


  @Override
  public BiasedLinear fit(VecDataSet learn, L2 l2) {
    final RVMCache cache = new RVMCache(learn.data(),l2.target,random);
    return cache.fit(tolerance);
  }
}