com.expleague.ml.models.multiclass.MultiClass2BinaryModel 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.Func;
import com.expleague.commons.math.MathTools;
import com.expleague.commons.math.vectors.Mx;
import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.util.logging.Logger;
import com.expleague.ml.models.MultiClassModel;

/**
 * User: solar
 * Date: 10.01.14
 * Time: 10:59
 */
@Deprecated
public class MultiClass2BinaryModel extends MultiClassModel {
  public static final Logger LOG = Logger.create(MultiClass2BinaryModel.class);
  private final Mx codingMatrix;
  private final Func[] binaryClassifiers;
  private final int dim;

  public MultiClass2BinaryModel(final Mx codingMatrix, final Func[] binaryClassifiers) {
    super(createStub(codingMatrix.rows(), binaryClassifiers[0].dim()));
    LOG.assertTrue(codingMatrix.columns() == binaryClassifiers.length, "Coding matrix columns count must match binary classifiers.");
//    final MxIterator mxIterator = codeMatrix.nonZeroes();
//    while (mxIterator.advance()) {
//      LOG.assertTrue(Math.abs(mxIterator.value()) < MathTools.EPSILON
//             || Math.abs(mxIterator.value() - 1.) < MathTools.EPSILON
//             || Math.abs(mxIterator.value() + 1.) < MathTools.EPSILON, "Coding matrix must contain elements from {-1,0,1} set.");
//    }
    this.codingMatrix = codingMatrix;
    this.binaryClassifiers = binaryClassifiers;
    dim = binaryClassifiers[0].dim();
//    for (int i = 0; i < dirs.length; i++) {
//      dirs[i] = new DecodeFunc(i);
//    }
  }

  private static Func[] createStub(final int count, final int dim) {
    final Func[] result = new Func[count];
    for (int i = 0; i < result.length; i++) {
      result[i] = new Func.Stub() {
        @Override
        public double value(final Vec x) {
          return 0;
        }

        @Override
        public int dim() {
          return dim;
        }
      };
    }
    return result;
  }

  private class DecodeFunc extends Func.Stub {
    private final int classNo;

    public DecodeFunc(final int classNo) {
      this.classNo = classNo;
    }

    @Override
    public double value(final Vec x) {
      double result = 0;
      for (int l = 0; l < codingMatrix.columns(); l++) {
        final double m = codingMatrix.get(l, classNo);
        if (Math.abs(m) < MathTools.EPSILON)
          continue;
        final double value = binaryClassifiers[l].value(x);
        result += value * m;
      }
      return result;
    }

    @Override
    public int dim() {
      return dim;
    }
  }
}