com.expleague.ml.models.multilabel.ThresholdProbsMultiLabelModel 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.multilabel;

import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.math.vectors.VecTools;
import com.expleague.ml.models.multiclass.MCModel;

/**
 * User: qdeee
 * Date: 22.03.15
 */
public class ThresholdProbsMultiLabelModel extends MultiLabelModel.Stub {
  private final MCModel intern;
  private final double threshold;

  public ThresholdProbsMultiLabelModel(final MCModel intern, final double threshold) {
    this.intern = intern;
    this.threshold = threshold;
  }

  @Override
  public Vec predictLabels(final Vec x) {
    final Vec probs = intern.probs(x);
    return VecTools.toBinary(probs, threshold);
  }

  @Override
  public int xdim() {
    return intern.xdim();
  }

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