All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.expleague.ml.models.multilabel.MultiLabelBinarizedModel Maven / Gradle / Ivy

There is a newer version: 1.4.9
Show newest version
package com.expleague.ml.models.multilabel;

import com.expleague.commons.math.Func;
import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.math.vectors.VecTools;
import com.expleague.ml.func.FuncJoin;

/**
 * User: qdeee
 * Date: 01.04.15
 */
public class MultiLabelBinarizedModel extends MultiLabelModel.Stub {
  private final FuncJoin model;

  public MultiLabelBinarizedModel(final FuncJoin model) {
    this.model = model;
  }

  public MultiLabelBinarizedModel(final Func[] funcs) {
    this.model = new FuncJoin(funcs);
  }

  public FuncJoin getInternModel() {
    return model;
  }

  @Override
  public Vec predictLabels(final Vec x) {
    final Vec trans = model.trans(x);
    return VecTools.toBinary(trans, 0);
  }

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

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

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

    final MultiLabelBinarizedModel that = (MultiLabelBinarizedModel) o;

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

    return true;
  }

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




© 2015 - 2024 Weber Informatics LLC | Privacy Policy