@@ -35,7 +35,7 @@ def read_pickle(a, f, i):
35
35
colors = prop_cycle .by_key ()['color' ]
36
36
37
37
n_trials = 4 # number of trials (independent experiments)
38
- algos = ['SPSO' , 'SPSOL' , 'CPSO' ,
38
+ algos = ['DEAPPSO' , ' SPSO' , 'SPSOL' , 'CPSO' ,
39
39
'CLPSO' , 'IPSO' , 'CCPSO2' ]
40
40
max_runtime , fitness_threshold = 3600.0 * 3 - 10.0 * 60 , 1e-10
41
41
# funcs = ['cigar', 'cigar_discus', 'ackley', 'bohachevsky',
@@ -95,14 +95,14 @@ def read_pickle(a, f, i):
95
95
# [yl.set_fontweight('bold') for yl in y_label]
96
96
97
97
lines , labels = axs [- 1 ][- 1 ].get_legend_handles_labels ()
98
- leg = fig .legend (lines , labels , loc = 'center' , ncol = 3 ,
98
+ leg = fig .legend (lines , labels , loc = 'center' , ncol = 4 ,
99
99
fontsize = font_size ,
100
100
bbox_to_anchor = (0.51 , 0.93 ))
101
101
# for text in leg.get_texts():
102
102
# text.set_fontweight('bold')
103
- # fig.text(0.05, 0.5, 'Fitness (Minimized)', va='center',
104
- # rotation='vertical', fontsize=font_size) # 'xx-large'
105
- # fig.text(0.5, 0.05, 'Runtime (Seconds)', va='center',
106
- # ha='center', fontsize=font_size) # 'xx-large'
103
+ fig .text (0.05 , 0.5 , 'Fitness (Minimized)' , va = 'center' ,
104
+ rotation = 'vertical' , fontsize = font_size ) # 'xx-large'
105
+ fig .text (0.5 , 0.05 , 'Runtime (Seconds)' , va = 'center' ,
106
+ ha = 'center' , fontsize = font_size ) # 'xx-large'
107
107
plt .savefig ('PSO.png' , dpi = 700 , bbox_inches = 'tight' )
108
108
plt .show ()
0 commit comments