Changeset 11749:cadf4246c059 in orange


Ignore:
Timestamp:
11/06/13 10:37:39 (5 months ago)
Author:
Ales Erjavec <ales.erjavec@…>
Branch:
default
Message:

PLS learner/predictor fix. Added tests.

Location:
Orange
Files:
1 added
1 edited

Legend:

Unmodified
Added
Removed
  • Orange/regression/pls.py

    r10580 r11749  
    243243        """      
    244244        domain = table.domain 
     245        multitarget = False 
    245246        if x_vars is None and y_vars is None: 
    246247            # Response variables are defined in the table. 
     
    250251            elif domain.class_vars: 
    251252                y_vars = domain.class_vars 
     253                multitarget = True 
    252254            else: 
    253255                raise TypeError('Class-less domain (x-vars and y-vars needed).') 
    254             x_table = select_attrs(table, x_vars) 
    255             y_table = select_attrs(table, y_vars) 
     256 
    256257        elif not (x_vars and y_vars): 
    257258            raise ValueError("Both x_vars and y_vars must be defined.") 
     259        else: 
     260            multitarget = True 
    258261 
    259262        x_table = select_attrs(table, x_vars) 
     
    272275         
    273276        domain = Orange.data.Domain(x_vars + y_vars, False) 
    274         multitarget = True if len(y_vars) > 1 else False 
    275277 
    276278        x = x_table.to_numpy()[0] 
     
    415417        self.domain = domain 
    416418        self.multitarget = multitarget 
     419        if multitarget and y_vars: 
     420            self.class_vars = y_vars 
     421        elif y_vars: 
     422            self.class_var = y_vars[0] 
     423 
    417424        self.coefs = coefs 
    418425        self.mu_x, self.mu_y = mu_x, mu_y 
Note: See TracChangeset for help on using the changeset viewer.