com.expleague.ml.func.Linear Maven / Gradle / Ivy

Go to download
package com.expleague.ml.func;

import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.math.vectors.VecTools;
import com.expleague.commons.math.vectors.impl.vectors.ArrayVec;
import com.expleague.commons.math.Func;
import com.expleague.commons.math.Trans;
import org.jetbrains.annotations.Nullable;

/**
 * User: solar
 * Date: 26.11.12
 * Time: 15:56
 */
public class Linear extends Func.Stub {
  public final Vec weights;

  public Linear(final double[] weights) {
    this(new ArrayVec(weights));
  }

  public Linear(final Vec weights) {
    this.weights = weights;
  }

  public Linear(final int size, final double step) {
    this.weights = VecTools.fill(new ArrayVec(size), step);
  }

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

  @Nullable
  @Override
  public Trans gradient() {
    return new Trans.Stub() {
      @Override
      public Vec trans(final Vec x) {
        return weights;
      }

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

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

  @Override
  public double value(final Vec point) {
    return VecTools.multiply(weights, point);
  }

  @Override
  public boolean equals(final Object o) {
    if (this == o) return true;
    if (!(o instanceof Linear)) return false;

    final Linear that = (Linear) o;
    return that.weights.equals(weights);
  }

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