@@ -239,6 +239,8 @@ def _generate_production_temperature_comparison_graph(
239239 output_path_stem : Path ,
240240 steady_state_start_years : float = _STEADY_STATE_START_YEARS ,
241241 geophires_data : pd .Series | None = None ,
242+ fervo_modeled_stats_caption : str = '' ,
243+ geophires_modeled_stats_caption : str = '' ,
242244) -> None :
243245 df_prod = pd .read_csv (production_csv_path )
244246 df_model = pd .read_csv (model_csv_path )
@@ -294,7 +296,7 @@ def _generate_production_temperature_comparison_graph(
294296 color = 'black' ,
295297 linestyle = '--' ,
296298 linewidth = 1.5 ,
297- label = 'Fervo -Modeled Temperature' if geophires_data is not None else 'Modeled output ' ,
299+ label = f' \n Fervo -Modeled Temperature{ fervo_modeled_stats_caption } ' ,
298300 )
299301
300302 if geophires_data is not None :
@@ -309,7 +311,7 @@ def _generate_production_temperature_comparison_graph(
309311 # (1, 3),
310312 # ), # loosely dotted - https://matplotlib.org/stable/gallery/lines_bars_and_markers/linestyles.html
311313 # linewidth=7,
312- label = 'GEOPHIRES-Modeled Temperature (Gringarten)' ,
314+ label = f 'GEOPHIRES-Modeled Temperature (Gringarten){ geophires_modeled_stats_caption } ' ,
313315 )
314316
315317 ax .set_xlabel ('Time (Years)' , fontsize = 12 )
@@ -424,17 +426,23 @@ def generate_fervo_project_red_2026_docs():
424426 r2_g = 1 - (ss_res_g / ss_tot ) if ss_tot != 0 else 0.0
425427
426428 _log .info (f'--- STATISTICAL ALIGNMENT (Steady-State > { _STEADY_STATE_START_YEARS } Years) ---' )
427- _log .info (f'FERVO: RMSE={ rmse_f :.2f} °C, R²={ r2_f :.4f} , Bias={ bias_f :.2f} °C' )
428- _log .info (f'GEOPHIRES: RMSE={ rmse_g :.2f} °C, R²={ r2_g :.4f} , Bias={ bias_g :.2f} °C' )
429+ fervo_modeled_stats_caption = f'RMSE={ rmse_f :.2f} °C, R²={ r2_f :.4f} , Bias={ bias_f :.2f} °C'
430+ geophires_modeled_stats_caption = f'RMSE={ rmse_g :.2f} °C, R²={ r2_g :.4f} , Bias={ bias_g :.2f} °C'
431+ _log .info (f'FERVO: { fervo_modeled_stats_caption } ' )
432+ _log .info (f'GEOPHIRES: { geophires_modeled_stats_caption } ' )
429433
430434 df_steady_state .to_csv (steady_state_csv_path , index = False )
431435
436+ _tab = ' '
437+
432438 _generate_production_temperature_comparison_graph (
433439 production_csv_path_ ,
434440 model_csv_path_ ,
435441 steady_state_csv_path ,
436442 generated_graph_path_stem ,
437443 geophires_data = geophires_series ,
444+ fervo_modeled_stats_caption = f'\n { _tab } { fervo_modeled_stats_caption } \n ' ,
445+ geophires_modeled_stats_caption = f'\n { _tab } { geophires_modeled_stats_caption } ' ,
438446 )
439447
440448
0 commit comments