print('>>> Ellapsed time: ', time.time() - time0)
# Graphics for Descriptive satatistics
__PLOTS = False # If True create plots subfolder under outputs folder
if __PLOTS:
from utils import plots
uparams_names = list(params.keys())
corep = read(os.path.join(cpo_dir, "ets_coreprof_in.cpo"), "coreprof")
rho = corep.rho_tor
uparams_names = list(params.keys())
test_case = cpo_dir.split('/')[-1]
plots.plot_stats_pctl(rho, stats_te, pctl_te,
xlabel=r'$\rho_{tor} ~ [m]$', ylabel=r'$Te$',
ftitle='Te profile ('+test_case+')',
fname='plots/Te_STAT_'+test_case)
plots.plot_sobols(rho, stot_te, uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Te',
fname='plots/Te_SA_'+test_case)
plots.plot_stats_pctl(rho, stats_ti, pctl_ti,
xlabel=r'$\rho_{tor} ~ [m]$', ylabel=r'$T_i [eV]$',
ftitle='Te profile ('+test_case+')',
fname='plots/Ti_STAT_'+test_case)
plots.plot_sobols(rho, stot_ti, uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Ti',
fname='plots/Ti_SA_'+test_case)
stats_ti = results['statistical_moments']['Ti']
pctl_ti = results['percentiles']['Ti']
stot_ti = results['sobols_total']['Ti']
print('>>> Ellapsed time: ', time.time() - time0)
# Graphics for Descriptive satatistics
print('>>> Statictics and SA plots')
corep = read(os.path.join(cpo_dir, "ets_coreprof_in.cpo"), "coreprof")
rho = corep.rho_tor
uparams_names = list(uncertain_params.keys())
plots.plot_stats_pctl(rho,
stats_te,
pctl_te,
xlabel=r'$\rho_{tor} ~ [m]$',
ylabel=r'$Te$',
ftitle='Te profile',
fname='plots/te_bnd_stats')
plots.plot_sobols(rho,
stot_te,
uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Te',
fname='plots/te_bnd_stot')
plots.plot_stats_pctl(rho,
stats_ti,
pctl_ti,
xlabel=r'$\rho_{tor} ~ [m]$',
ylabel=r'$T_i [eV]$',
stats_ti = results['statistical_moments']['Ti']
pctl_ti = results['percentiles']['Ti']
stot_ti = results['sobols_total']['Ti']
print('>>> Ellapsed time: ', time.time() - time0)
# Graphics for Sescriptive satatistics
print('>>> Statictics and SA plots')
corep = read(os.path.join(cpo_dir, "ets_coreprof_in.cpo"), "coreprof")
rho = corep.rho_tor
uparams_names = list(uncertain_params.keys())
plots.plot_stats_pctl(rho,
stats_te,
pctl_te,
xlabel=r'$\rho_{tor} ~ [m]$',
ylabel=r'$Te$',
ftitle='Te profile',
fname='outputs/figs/te_src_stat')
plots.plot_sobols(rho,
stot_te,
uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Te',
fname='outputs/figs/te_src_sob')
plots.plot_stats_pctl(rho,
stats_ti,
pctl_ti,
xlabel=r'$\rho_{tor} ~ [m]$',
ylabel=r'$T_i [eV]$',
sob1_ti_df.to_sql('SOB1_TI', engine, if_exists='append')
sobt_ti_df = pd.DataFrame.from_dict(sobt_ti)
sobt_ti_df.to_sql('SOBT_TI', engine, if_exists='append')
os.system('cp ' + engine.url.database + ' outputs')
# Plots STAT and SA
__PLOTS = True # If True create plots subfolder under outputs folder
if __PLOTS:
from utils import plots
uparams_names = list(params.keys())
plots.plot_stats_pctl(rho,
stat_te,
pctl_te,
xlabel=r'$\rho_{tor} ~ [m]$',
ylabel=r'$Te$',
ftitle='Te profile (Ions S) - ' + test_case,
fname='outputs/plots/Te_STAT_Ion_' + test_case)
plots.plot_sobols(rho,
sobt_te,
uparams_names,
ftitle=' Total-Order Sobol indices (Ion S)- QoI: Te',
fname='outputs/plots/Te_SA_Ion_' + test_case)
plots.plot_stats_pctl(rho,
stat_ti,
pctl_ti,
xlabel=r'$\rho_{tor} ~ [m]$',
ylabel=r'$T_i [eV]$',