Update embedding tag

frontend/src/high-dimensional/HighDimensional.vue

@@ -12,6 +12,7 @@
     <div class="visual-dl-page-right">
       <div class="visual-dl-page-config-container">
         <ui-config
+          :runs-items="runsItems"
           :config="config"
         />
       </div>
@@ -20,7 +21,7 @@
 </template>
 
 <script>
-import {getHighDimensionalDatasets} from '../service';
+import {getHighDimensionalDatasets, getRuns} from '../service';
 import autoAdjustHeight from '../common/util/autoAdjustHeight';
 import Config from './ui/Config';
 import Chart from './ui/Chart';
@@ -33,18 +34,27 @@ export default {
   name: 'HighDimensional',
   data() {
     return {
+      runsArray: [],
       config: {
         searchText: '',
         displayWordLabel: true,
         dimension: '2',
         reduction: 'tsne',
+        showingRun: '',
         running: true,
       },
       embeddingData: [],
     };
   },
   created() {
-    this.fetchDatasets();
+    getRuns().then(({errno, data}) => {
+      this.runsArray = data;
+
+      // Setting showingRun should trigger fetchDatasets
+      if (data.length > 0) {
+        this.config.showingRun = data[0];
+      }
+    });
   },
   watch: {
     'config.dimension': function(val) {
@@ -53,16 +63,31 @@ export default {
     'config.reduction': function(val) {
       this.fetchDatasets();
     },
+    'config.showingRun': function(val) {
+      this.fetchDatasets();
+    },
   },
   mounted() {
     autoAdjustHeight();
   },
+  computed: {
+    runsItems() {
+      let runsArray = this.runsArray || [];
+      return runsArray.map((item) => {
+        return {
+          name: item,
+          value: item,
+        };
+      });
+    },
+  },
   methods: {
     fetchDatasets() {
       // Fetch the data from the server. Passing dimension and reduction method
       let params = {
         dimension: this.config.dimension,
         reduction: this.config.reduction,
+        run: this.config.showingRun,
       };
       getHighDimensionalDatasets(params).then(({errno, data}) => {
         let vectorData = data.embedding;

frontend/src/high-dimensional/ui/Config.vue

@@ -39,6 +39,17 @@
         value="pca"/>
     </v-radio-group>
 
+    <v-radio-group
+      label="Run"
+      v-model="config.showingRun"
+      dark>
+      <v-radio
+        v-for="item in runsItems"
+        :key="item.name"
+        :label="item.name"
+        :value="item.value" />
+    </v-radio-group>
+
     <v-btn
       :color="config.running ? 'primary' : 'error'"
       v-model="config.running"
@@ -55,6 +66,10 @@
 
 export default {
   props: {
+    runsItems: {
+      type: Array,
+      required: true,
+    },
     config: {
       type: Object,
       required: true,

visualdl/python/storage.py

@@ -18,10 +18,12 @@
 from visualdl import core
 
 dtypes = ("float", "double", "int32", "int64")
+EMBEDDING_TAG = 'embedding'
 
 
 def check_tag_name_valid(tag):
     assert '%' not in tag, "character % is a reserved word, it is not allowed in tag."
+    assert tag != EMBEDDING_TAG, "embedding is a reserved word, it is not allowed in tag."
 
 
 def check_mode_name_valid(tag):
@@ -140,9 +142,8 @@ def text(self, tag):
         check_tag_name_valid(tag)
         return self.reader.get_text(tag)
 
-    def embedding(self, tag):
-        check_tag_name_valid(tag)
-        return self.reader.get_embedding(tag)
+    def embedding(self):
+        return self.reader.get_embedding(EMBEDDING_TAG)
 
     def audio(self, tag):
         """
@@ -290,9 +291,8 @@ def text(self, tag):
         check_tag_name_valid(tag)
         return self.writer.new_text(tag)
 
-    def embedding(self, tag):
-        check_tag_name_valid(tag)
-        return self.writer.new_embedding(tag)
+    def embedding(self):
+        return self.writer.new_embedding(EMBEDDING_TAG)
 
     def save(self):
         self.writer.save()

visualdl/server/lib.py

@@ -303,14 +303,9 @@ def get_texts(storage, mode, tag, num_records=100):
         return res
 
 
-def get_embeddings(storage,
-                   mode,
-                   tag,
-                   reduction,
-                   dimension=2,
-                   num_records=5000):
+def get_embeddings(storage, mode, reduction, dimension=2, num_records=5000):
     with storage.mode(mode) as reader:
-        embedding = reader.embedding(tag)
+        embedding = reader.embedding()
         labels = embedding.get_all_labels()
         high_dimensional_vectors = embedding.get_all_embeddings()
 

visualdl/server/visualDL

@@ -261,10 +261,11 @@ def texts():
 
 @app.route('/data/plugin/embeddings/embeddings')
 def embeddings():
+    run = request.args.get('run')
     dimension = request.args.get('dimension')
     reduction = request.args.get('reduction')
-    key = os.path.join('/data/plugin/embeddings/embeddings', dimension, reduction)
-    data = cache_get(key, try_call, lib.get_embeddings, log_reader, 'train', 'scratch/embedding', reduction, int(dimension))
+    key = os.path.join('/data/plugin/embeddings/embeddings', run, dimension, reduction)
+    data = cache_get(key, try_call, lib.get_embeddings, log_reader, run, reduction, int(dimension))
     result = gen_result(0, "", data)
     return Response(json.dumps(result), mimetype='application/json')