Spaces:
No application file
No application file
import dot from 'compute-dot'; | |
import cosineSimilarity from 'compute-cosine-similarity'; | |
import { getSimilarityMeasure } from '../config'; | |
const computeSimilarity = (x: number[], y: number[]): number => { | |
const similarityMeasure = getSimilarityMeasure(); | |
if (similarityMeasure === 'cosine') { | |
return cosineSimilarity(x, y); | |
} else if (similarityMeasure === 'dot') { | |
return dot(x, y); | |
} | |
throw new Error('Invalid similarity measure'); | |
}; | |
export default computeSimilarity; | |