__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)
print('>>> ets_uq : END')
# 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]$',
ftitle='Te profile',
fname='plots/ti_bnd_stats')
plots.plot_sobols(rho,
stot_ti,
uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Ti',
# 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]$',
ftitle='Te profile',
fname='outputs/figs/ti_src_stat')
plots.plot_sobols(rho,
stot_ti,
uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Ti',
__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]$',
ftitle='Ti profile - (Ions S) - ' + test_case,
fname='outputs/plots/Ti_STAT_Ion_' + test_case)
plots.plot_sobols(rho,
sobt_ti,
uparams_names,
ftitle=' Total-Order Sobol indices - QoI: Ti',