package org.apache.mahout.benchmark;
import static org.apache.mahout.benchmark.VectorBenchmarks.DENSE_FN_RAND;
import static org.apache.mahout.benchmark.VectorBenchmarks.DENSE_FN_SEQ;
import static org.apache.mahout.benchmark.VectorBenchmarks.DENSE_VECTOR;
import static org.apache.mahout.benchmark.VectorBenchmarks.RAND_FN_DENSE;
import static org.apache.mahout.benchmark.VectorBenchmarks.RAND_FN_SEQ;
import static org.apache.mahout.benchmark.VectorBenchmarks.RAND_SPARSE_VECTOR;
import static org.apache.mahout.benchmark.VectorBenchmarks.SEQ_FN_DENSE;
import static org.apache.mahout.benchmark.VectorBenchmarks.SEQ_FN_RAND;
import static org.apache.mahout.benchmark.VectorBenchmarks.SEQ_SPARSE_VECTOR;
import org.apache.mahout.benchmark.BenchmarkRunner.BenchmarkFnD;
import org.apache.mahout.common.distance.DistanceMeasure;
public class DistanceBenchmark {
private final VectorBenchmarks mark;
public DistanceBenchmark(VectorBenchmarks mark) {
this.mark = mark;
}
public void benchmark(final DistanceMeasure measure) {
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[0][mark.vIndex(i)], mark.vectors[0][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), DENSE_VECTOR);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[1][mark.vIndex(i)], mark.vectors[1][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), RAND_SPARSE_VECTOR);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[2][mark.vIndex(i)], mark.vectors[2][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), SEQ_SPARSE_VECTOR);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[0][mark.vIndex(i)], mark.vectors[1][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), DENSE_FN_RAND);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[0][mark.vIndex(i)], mark.vectors[2][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), DENSE_FN_SEQ);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[1][mark.vIndex(i)], mark.vectors[0][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), RAND_FN_DENSE);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[1][mark.vIndex(i)], mark.vectors[2][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), RAND_FN_SEQ);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[2][mark.vIndex(i)], mark.vectors[0][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), SEQ_FN_DENSE);
mark.printStats(mark.getRunner().benchmarkD(new BenchmarkFnD() {
@Override
public Double apply(Integer i) {
return measure.distance(mark.vectors[2][mark.vIndex(i)], mark.vectors[1][mark.vIndex(randIndex())]);
}
}), measure.getClass().getName(), SEQ_FN_RAND);
}
}