com.expleague.ml.models.ShifftedTrans Maven / Gradle / Ivy

Go to download

package com.expleague.ml.models;

import com.expleague.commons.math.vectors.Vec;
import com.expleague.commons.math.Trans;

import static com.expleague.commons.math.vectors.VecTools.adjust;


public class ShifftedTrans extends Trans.Stub {
  private final double mean;
  private final Trans trans;
  public ShifftedTrans(Trans result, double v) {
    this.trans = result;
    this.mean = v;
  }

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

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

  @Override
  public Vec trans(Vec x) {
    final Vec res = trans.trans(x);
    return adjust(res, mean);
  }

}