File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -848,7 +848,7 @@ def calculate_log_reg_coefficients(self):
848848 if len (self .domain .class_var .values ) == 2 :
849849 self .b0 = np .hstack ((self .b0 * (- 1 ), self .b0 ))
850850 coeffs = np .vstack ((coeffs * (- 1 ), coeffs ))
851- self .log_reg_coeffs = [coeffs [:, ranges [ i ]] for i in range ( len ( attrs )) ]
851+ self .log_reg_coeffs = [coeffs [:, r ] for r in ranges ]
852852 self .log_reg_coeffs_orig = self .log_reg_coeffs .copy ()
853853
854854 min_values = nanmin (self .data .X , axis = 0 )
@@ -860,7 +860,7 @@ def calculate_log_reg_coefficients(self):
860860 coef = self .log_reg_coeffs [i ]
861861 self .log_reg_coeffs [i ] = np .hstack ((coef * min_t , coef * max_t ))
862862 self .log_reg_cont_data_extremes .append (
863- [sorted ([min_t , max_t ], reverse = (c < 0 )) for c in coef ])
863+ [sorted ([min_t , max_t ], reverse = (c < 0 )) for c in coef . flat ])
864864 else :
865865 self .log_reg_cont_data_extremes .append ([None ])
866866
You can’t perform that action at this time.
0 commit comments