org.deeplearning4j.distancefunction.CosineSimilarity Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deeplearning4j-core Show documentation

There is a newer version: 1.0.0-M2.1

package org.deeplearning4j.distancefunction;

import org.deeplearning4j.util.MatrixUtil;
import org.jblas.DoubleMatrix;


public class CosineSimilarity extends BaseDistanceFunction {

	
	private static final long serialVersionUID = 3272217249919443730L;

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

	@Override
	public Double apply(DoubleMatrix input) {
		return MatrixUtil.cosineSim(input, base);
	}

	
	
}