@@ -537,10 +537,20 @@ def print_results(self):
537537 4. min mean X
538538 """
539539 print (f"min y: { self .min_y } " )
540- print (f"min X: { self .to_all_dim (self .min_X .reshape (1 , - 1 ))} " )
540+ res = self .to_all_dim (self .min_X .reshape (1 , - 1 ))
541+ for i in range (res .shape [1 ]):
542+ if self .var_name is None :
543+ print ("x" + str (i ) + ":" , res [0 ][i ])
544+ else :
545+ print (self .var_name [i ] + ":" , res [0 ][i ])
541546 if self .noise :
547+ res = self .to_all_dim (self .min_mean_X .reshape (1 , - 1 ))
542548 print (f"min mean y: { self .min_mean_y } " )
543- print (f"min mean X: { self .to_all_dim (self .min_mean_X .reshape (1 , - 1 ))} " )
549+ for i in range (res .shape [1 ]):
550+ if self .var_name is None :
551+ print ("x" + str (i ) + ":" , res [0 ][i ])
552+ else :
553+ print (self .var_name [i ] + ":" , res [0 ][i ])
544554
545555 def chg (self , x , y , z0 , i , j ):
546556 z0 [i ] = x
@@ -598,5 +608,9 @@ def print_importance(self):
598608 theta = np .power (10 , self .surrogate .theta )
599609 print ("Importance relative to the most important parameter:" )
600610 imp = 100 * theta / np .max (theta )
601- for i in range (len (imp )):
602- print ("Parameter" , i , ": " , imp [i ])
611+ if self .var_name is None :
612+ for i in range (len (imp )):
613+ print ("x" , i , ": " , imp [i ])
614+ else :
615+ for i in range (len (imp )):
616+ print (self .var_name [i ] + ": " , imp [i ])
0 commit comments