Merge pull request #173 from AirGrid/release/v0.0.0-dev.40

release: v0.0.0-dev.40 to MASTER
AirGrid · Feb 18, 2022 · 7653842 · 7653842
2 parents 2348f7e + 830f69e
commit 7653842
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 10 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@airgrid/edgekit",
-  "version": "0.0.0-dev.39",
+  "version": "0.0.0-dev.40",
   "homepage": "https://edgekit.org/",
   "author": "AirGrid <https://airgrid.io>",
   "license": "MIT",

diff --git a/src/engine/filters/math.ts b/src/engine/filters/math.ts
@@ -1,19 +1,28 @@
-export const dotProduct = (vectorA: number[], vectorB: number[]): number => {
-  return vectorA.reduce((acc, cur, idx) => acc + cur * vectorB[idx], 0);
-};
-
 const euclideanLength = (vector: number[]): number => {
   return Math.sqrt(vector.reduce((acc, cur) => acc + Math.pow(cur, 2), 0));
 };
 
+const cosineScaler = (x: number): number => {
+  const MAX = 0.9;
+  const MIN = -0.5;
+  const scaled = (x - MIN) / (MAX - MIN);
+  if (scaled >= 1) return 1;
+  if (scaled <= 0) return 0;
+  return scaled;
+};
+
+export const dotProduct = (vectorA: number[], vectorB: number[]): number => {
+  return vectorA.reduce((acc, cur, idx) => acc + cur * vectorB[idx], 0);
+};
+
 export const cosineSimilarity = (
   vectorA: number[],
   vectorB: number[]
 ): number => {
-  return (
+  const cosineSim =
     dotProduct(vectorA, vectorB) /
-    (euclideanLength(vectorA) * euclideanLength(vectorB))
-  );
+    (euclideanLength(vectorA) * euclideanLength(vectorB));
+  return cosineScaler(cosineSim);
 };
 
 export const sigmoid = (z: number): number => {