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]$',