com.expleague.ml.models.multiclass.JoinedBinClassModel 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.models.multiclass;

import com.expleague.commons.math.MathTools;
import com.expleague.commons.math.vectors.impl.vectors.ArrayVec;
import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.util.ArrayTools;
import com.expleague.commons.math.Func;
import com.expleague.ml.func.FuncJoin;

/**
 * User: qdeee
 * Date: 14.04.14
 * Description: this class is similar to MultiClassModel, however for JoinedBinClassModel it's assuming that you have
 * pack of INDEPENDENT binary classifier and their count equals to classes count.
 */
public class JoinedBinClassModel extends MCModel.Stub {
  protected final FuncJoin internalModel;

  public JoinedBinClassModel(final Func[] dirs) {
    internalModel = new FuncJoin(dirs);
  }

  public JoinedBinClassModel(final FuncJoin internalModel) {
    this.internalModel = internalModel;
  }

  public FuncJoin getInternModel() {
    return internalModel;
  }

  @Override
  public int countClasses() {
    return internalModel.ydim();
  }

  @Override
  public Vec probs(final Vec x) {
    final Vec apply = internalModel.trans(x);
    final Vec probs = new ArrayVec(apply.dim());
    for (int i = 0; i < apply.dim(); i++) {
      probs.set(i, MathTools.sigmoid(apply.get(i)));
    }
    return probs;
  }

  @Override
  public int bestClass(final Vec x) {
    final double[] trans = internalModel.trans(x).toArray();
    return ArrayTools.max(trans);
  }

  @Override
  public int dim() {
    return internalModel.dirs[0].xdim();
  }

  @Override
  public boolean equals(final Object o) {
    if (this == o) return true;
    if (o == null || getClass() != o.getClass()) return false;

    final JoinedBinClassModel that = (JoinedBinClassModel) o;

    if (!internalModel.equals(that.internalModel)) return false;

    return true;
  }

  @Override
  public int hashCode() {
    return internalModel.hashCode();
  }
}