com.expleague.ml.dynamicGrid.impl.BinaryFeatureImpl Maven / Gradle / Ivy

Go to download
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;
}