bookmark

Author: alicialics

tensorboard/plugins/projector/vz_projector/data.ts

@@ -656,6 +656,9 @@ export class State {
   tSNEIteration: number = 0;
   tSNEPerplexity: number = 0;
   tSNELearningRate: number = 0;
+  tSNESuperviseFactor: number = 0;
+  tSNESuperviseInput: string;
+  tSNESuperviseColumn: string;
   tSNEis3d: boolean = true;
   /** UMAP parameters */
   umapIs3d: boolean = true;

tensorboard/plugins/projector/vz_projector/vz-projector-data-panel.ts

@@ -17,12 +17,7 @@ import {PolymerElement} from '@polymer/polymer';
 import '../../../components/polymer/irons_and_papers';
 import {LegacyElementMixin} from '../../../components/polymer/legacy_element_mixin';
 import * as d3 from '../../../webapp/third_party/d3';
-import {
-  ColorOption,
-  ColumnStats,
-  Projection,
-  SpriteAndMetadataInfo,
-} from './data';
+import {ColorOption, ColumnStats, SpriteAndMetadataInfo} from './data';
 import {
   DataProvider,
   EmbeddingInfo,

tensorboard/plugins/projector/vz_projector/vz-projector-projections-panel.ts

@@ -266,6 +266,9 @@ class ProjectionsPanel extends LegacyElementMixin(PolymerElement) {
     if (this.learningRateInput) {
       this.learningRateInput.value = bookmark.tSNELearningRate.toString();
     }
+    this.superviseFactor = bookmark.tSNESuperviseFactor;
+    this.superviseColumn = bookmark.tSNESuperviseColumn;
+    this.superviseInput = bookmark.tSNESuperviseInput || '';
     this.tSNEis3d = bookmark.tSNEis3d;
     // UMAP
     this.umapIs3d = bookmark.umapIs3d;
@@ -301,6 +304,9 @@ class ProjectionsPanel extends LegacyElementMixin(PolymerElement) {
     this.setZDropdownEnabled(this.pcaIs3d);
     this.updateTSNEPerplexityFromSliderChange();
     this.updateTSNELearningRateFromUIChange();
+    this.updateTSNESuperviseFactorFromUIChange();
+    this.setSupervision(this.superviseInput, this.superviseColumn);
+    this.superviseInputChange();
     if (this.iterationLabelTsne) {
       this.iterationLabelTsne.innerText = bookmark.tSNEIteration.toString();
     }
@@ -323,6 +329,9 @@ class ProjectionsPanel extends LegacyElementMixin(PolymerElement) {
     if (this.learningRateInput != null) {
       bookmark.tSNELearningRate = +this.learningRateInput.value;
     }
+    bookmark.tSNESuperviseFactor = this.superviseFactor;
+    bookmark.tSNESuperviseInput = this.superviseInput;
+    bookmark.tSNESuperviseColumn = this.superviseColumn;
     bookmark.tSNEis3d = this.tSNEis3d;
     // UMAP
     bookmark.umapIs3d = this.umapIs3d;

tensorboard/plugins/projector/vz_projector/vz-projector.ts

@@ -672,7 +672,6 @@ class Projector
         point.projections[keys[j]] = projection[keys[j]];
       }
     }
-    this.dataSet.hasTSNERun = state.selectedProjection === 'tsne';
     this.dataSet.tSNEIteration = state.tSNEIteration;
     this.projectionsPanel.restoreUIFromBookmark(state);
     this.inspectorPanel.restoreUIFromBookmark(state);