package org.nd4j.linalg.distancefunction;

import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.ops.transforms.Transforms;


public class CosineSimilarity extends BaseDistanceFunction {

	
	private static final long serialVersionUID = 3272217249919443730L;

	public CosineSimilarity(INDArray base) {
		super(base);
	}

	@Override
	public Float apply(INDArray input) {
		return (float) Transforms.cosineSim(input, base);
	}

	
	
}