Merge pull request #5839 from VesnaT/effective_data

[FIX] tSNE, Freeviz: Set effective_data

Author: lanzagar

Orange/widgets/unsupervised/owtsne.py

@@ -291,6 +291,10 @@ def __init__(self):
         self.tsne_embedding = None  # type: Optional[manifold.TSNEModel]
         self.iterations_done = 0    # type: int
 
+    @property
+    def effective_data(self):
+        return self.data.transform(Domain(self.effective_variables))
+
     def _add_controls(self):
         self._add_controls_start_box()
         super()._add_controls()

Orange/widgets/visualize/owfreeviz.py

@@ -9,7 +9,7 @@
 
 import pyqtgraph as pg
 
-from Orange.data import Table
+from Orange.data import Table, Domain
 from Orange.projection import FreeViz
 from Orange.projection.freeviz import FreeVizModel
 from Orange.widgets import widget, gui, settings
@@ -181,6 +181,11 @@ def effective_variables(self):
         return [a for a in self.data.domain.attributes
                 if a.is_continuous or a.is_discrete and len(a.values) == 2]
 
+    @property
+    def effective_data(self):
+        return self.data.transform(Domain(self.effective_variables,
+                                          self.data.domain.class_vars))
+
     def __radius_slider_changed(self):
         self.graph.update_radius()