Skip to content

Commit a5d3ef4

Browse files
include stats alignment metrics in captions
1 parent 2df4bd8 commit a5d3ef4

3 files changed

Lines changed: 12 additions & 4 deletions
13 KB
Loading
13.5 KB
Loading

src/geophires_docs/generate_fervo_project_red_2026_docs.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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'\nFervo-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

Comments
 (0)