source: branches/collapse/SEMILOG_PLOTS.py @ 855

#!/usr/bin/env python
#
# Author: Lan Huong Nguyen (lanhuong @stanford)
# Copyright (c) 2012-2016 California Institute of Technology.
# License: 3-clause BSD.  The full license text is available at:
#  - http://mmckerns.github.io/project/mystic/browser/mystic/LICENSE

def my_read_paramlog_file(file):
    from mystic.munge import read_support_file
    steps, cost_hist  = read_support_file(file)
    
    params_hist = []
    
    generations =  len(steps[0][0])
    nDim = len(steps[0])
    
    for n in range(generations):
        c = []
        for i in range(nDim):
            c.append(steps[0][i][n])
        params_hist.append(c)
    
    return params_hist, cost_hist

p_col_222,c_col_222 = my_read_paramlog_file("paramlog_collapse_222_ngen200_ngcol100_3.py")
p_col_333,c_col_333 = my_read_paramlog_file("paramlog_collapse_333_ngen200_ngcol100_4.py")
p_col_555,c_col_555 = my_read_paramlog_file("paramlog_collapse_555_ngen200_ngcol100_2.py")

p_222,c_222 = my_read_paramlog_file("paramlog_COG_222_ngen200_5.py")
p_333,c_333 = my_read_paramlog_file("paramlog_COG_333_ngen200_3.py")
p_555,c_555 = my_read_paramlog_file("paramlog_COG_555_ngen200_2.py")


'''print len(cost_hist)
print len(cost_hist2)'''

import numpy as np
import matplotlib.pyplot as plt
from math import log10

x_col_222 = np.arange(0, len(c_col_222), 1)
y_col_222 = [log10(abs(-1*ele-0.378970568266)) for ele in c_col_222]

x_col_333 = np.arange(0, len(c_col_333), 1)
y_col_333 = [log10(abs(-1*ele-0.378970568266)) for ele in c_col_333]

x_col_555 = np.arange(0, len(c_col_555), 1)
y_col_555 = [log10(abs(-1*ele-0.378970568266)) for ele in c_col_555]

x_222 = np.arange(0, len(c_222), 1)
y_222= [log10(abs(-1*ele-0.378970568266)) for ele in c_222]

x_333 = np.arange(0, len(c_333), 1)
y_333= [log10(abs(-1*ele-0.378970568266)) for ele in c_333]

x_555 = np.arange(0, len(c_555), 1)
y_555 = [log10(abs(-1*ele-0.378970568266)) for ele in c_555]


plt.plot(x_col_222, y_col_222, 'b:', x_col_333, y_col_333, 'b--', x_col_555, \
         y_col_555, 'b-', x_222, y_222, 'g:', x_333, y_333, 'g--', x_555, \
         y_555, 'g-', linewidth = 2)

'''
#npts 555, 2
plt.axvline(x=101, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=147, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=370, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=488, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=674, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=1132, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=1311, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=2488, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')

#npts 333, 2
plt.axvline(x=145, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=530, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=785, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')


#npts 222, 3
plt.axvline(x=260, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=510, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')

#npts 222, 1
plt.axvline(x=255, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=600, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')

#npts 222, 2
plt.axvline(x=220, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')

#npts 333, 4
plt.axvline(x=120, ymin=-10, ymax=1, hold = None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=125, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=270, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=490, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=740, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')

# npts 333, 5
plt.axvline(x=139, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
#plt.axvline(x=249, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=388, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=556, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
#plt.axvline(x=629, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=717, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=824, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
#plt.axvline(x=930, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')
plt.axvline(x=1238, ymin=-10, ymax=1, hold=None, linestyle = '--', linewidth=1, color='r')'''

plt.legend(('COG & CLPS, 222', 'COG & CLPS, 333', 'COG & CLPS, 555', 'COG, 222', \
            'COG, 333', 'COG, 555'), 'lower right', shadow=True, fancybox=True)

plt.title('Semilog Plot for Convergence Rate')
plt.xlabel('iteration number, n')
plt.ylabel('$log-error,\; log_{10}(\hat{P} - \hat{P}_{max})$')
plt.grid(True)
plt.show()