Embedding Projector: perform consistent knn calculation

tensorflow · Apr 27, 2023 · 811f3d0 · 811f3d0
1 parent b72d751
commit 811f3d0
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/tensorboard/plugins/projector/vz_projector/data.ts b/tensorboard/plugins/projector/vz_projector/data.ts
@@ -86,7 +86,7 @@ export interface DataPoint {
     [key: string]: number;
   };
 }
-const IS_FIREFOX = navigator.userAgent.toLowerCase().indexOf('firefox') >= 0;
+
 /** Maximum sample size for each projection type. */
 export const TSNE_SAMPLE_SIZE = 10000;
 export const UMAP_SAMPLE_SIZE = 5000;
@@ -470,23 +470,15 @@ export class DataSet {
           .map((neighbors) => neighbors.slice(0, nNeighbors))
       );
     } else {
-      const knnGpuEnabled = (await util.hasWebGLSupport()) && !IS_FIREFOX;
       const numKnnNeighborsToCompute = Math.max(
         nNeighbors,
         MIN_NUM_KNN_NEIGHBORS
       );
-      const result = await (knnGpuEnabled
-        ? knn.findKNNGPUCosDistNorm(
-            data,
-            numKnnNeighborsToCompute,
-            (d) => d.vector
-          )
-        : knn.findKNN(
-            data,
-            numKnnNeighborsToCompute,
-            (d) => d.vector,
-            (a, b) => vector.cosDistNorm(a, b)
-          ));
+      const result = await knn.findKNNGPUCosDistNorm(
+        data,
+        numKnnNeighborsToCompute,
+        (d) => d.vector
+      );
       this.nearest = result;
       return Promise.resolve(
         result.map((neighbors) => neighbors.slice(0, nNeighbors))