Skip to content

Commit 84c3c21

Browse files
committed
owlinearprojection: Add class to effective_data for LDA.
1 parent 3516fc3 commit 84c3c21

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

Orange/widgets/visualize/owlinearprojection.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,10 @@ def effective_variables(self):
329329

330330
@property
331331
def effective_data(self):
332-
return self.data.transform(Domain(self.effective_variables))
332+
cvs = None
333+
if self.placement == Placement.LDA:
334+
cvs = self.data.domain.class_vars
335+
return self.data.transform(Domain(self.effective_variables, cvs))
333336

334337
def __vizrank_set_attrs(self, attrs):
335338
if not attrs:

Orange/widgets/visualize/tests/test_owlinearprojection.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,8 @@ def test_no_data_for_lda(self):
8585
self.send_signal(self.widget.Inputs.data, self.data)
8686
self.widget.radio_placement.buttons[Placement.LDA].click()
8787
self.assertTrue(buttons[Placement.LDA].isEnabled())
88+
output = self.get_output(self.widget.Outputs.components)
89+
self.assertTrue(output and len(output) == 2)
8890
self.send_signal(self.widget.Inputs.data, Table("housing"))
8991
self.assertFalse(buttons[Placement.LDA].isEnabled())
9092
self.send_signal(self.widget.Inputs.data, None)

0 commit comments

Comments
 (0)