com.expleague.ml.dynamicGrid.impl.BinaryFeatureImpl 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.dynamicGrid.impl;

import com.expleague.ml.dynamicGrid.interfaces.BinaryFeature;
import com.expleague.ml.dynamicGrid.interfaces.DynamicRow;
import com.expleague.commons.math.vectors.Vec;

import java.util.Comparator;

/**
 * Created by noxoomo on 23/07/14.
 */
public class BinaryFeatureImpl implements BinaryFeature {
  public final DynamicRow bfRow;
  private int binNo;
  public final int origFIndex;
  public final int borderIndex;


  public final double condition;
  private boolean active = false;
  private double regScore = 0;

  public void setRegScore(final double reg) {
    this.regScore = reg;
  }


  public BinaryFeatureImpl(final DynamicRow bfRow, final int origFIndex, final double condition, final int index) {
    this.bfRow = bfRow;
    this.origFIndex = origFIndex;
    this.condition = condition;
    this.borderIndex = index;
  }


  public void setBinNo(final int newBinNo) {
    this.binNo = newBinNo;
  }


  @Override
  public void setActive(final boolean status) {
    this.active = status;
  }

  @Override
  public double regularization() {
    if (active) return 0;
    return regScore;
  }

  @Override
  public double condition() {
    return this.condition;
  }

  @Override
  public boolean value(final Vec vec) {
    return vec.get(origFIndex) > condition;
  }

  @Override
  public DynamicRow row() {
    return bfRow;
  }

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

  @Override
  public int fIndex() {
    return origFIndex;
  }


  @Override
  public boolean isActive() {
    return active;
  }


//    @Override
//    public boolean equals(Object o) {
//        if (this == o) return true;
//        if (!(o instanceof BinaryFeatureImpl)) return false;
//        BinaryFeatureImpl that = (BinaryFeatureImpl) o;
//        return fIndex() == that.fIndex() && bfRow.equals(that.bfRow);
//    }

//    @Override
//    public int hashCode() {
//        int result = bfRow.hashCode();
//        result = 31 * result;
//        return result;
//    }

  @Override
  public String toString() {
    return String.format("f[%d] > %g", origFIndex, condition);
  }


  static Comparator conditionComparator = new Comparator() {
    @Override
    public int compare(final BinaryFeatureImpl a, final BinaryFeatureImpl b) {
      return Double.compare(a.condition, b.condition);
    }
  };

  static Comparator borderComparator = new Comparator() {
    @Override
    public int compare(final BinaryFeatureImpl a, final BinaryFeatureImpl b) {
      return Integer.compare(a.borderIndex, b.borderIndex);
    }
  };


  public int gridHash;
}