# Provenance:
#   Script: /media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/scripts/t18_joint_embed_to_productivity_20260127_200000.py
#   Input: /media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_env.npy, /media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_pfam_modules.npy, /media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_bio_response.npy, /media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_bio_valid.npy
#   Models: /media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/models/world_model_fold_*_20260127_111754.pt
#   Date: 2026-01-27 12:01:32
#   Integrity Check: PASSED
#   N_samples: 1810 (bio_valid=1151)
#   Model: Frozen Encoder_E + Encoder_P (latent_dim=32) -> NEW Predictor (64D -> 3)
#   Method: [z_env, z_pfam] -> Predictor -> productivity
#   CV: Leave-one-basin-out (6 folds, Red_Sea merged into Indian)
#   Runtime: 3.9s
fold	target	n_test	r2	rmse	mae
Arctic	chl_mean_mg_m3	28	-7.302507795602123	1.270542461446279	1.1658901913142483
Arctic	poc_mean_mg_m3	28	-4.793171291890494	105.51750793613284	95.8071696343053
Arctic	nflh_mean	28	-0.08352333892669406	0.03938386637689893	0.030229068480163213
Atlantic	chl_mean_mg_m3	363	0.5720330258430064	4.135122945389252	1.6796343244673795
Atlantic	poc_mean_mg_m3	363	0.6585752254159549	302.77636683468927	142.08932882150089
Atlantic	nflh_mean	363	0.6275742946668956	0.0491095451614236	0.03164230535107841
Indian	chl_mean_mg_m3	163	-0.8586721288178369	0.9541774523060301	0.694509446950703
Indian	poc_mean_mg_m3	163	0.1537531909845955	101.57495764212354	75.40090832903248
Indian	nflh_mean	163	0.28278750456985813	0.05096459332146571	0.02737724389707266
Mediterranean	chl_mean_mg_m3	127	-4.619533470498027	1.53128843577843	1.2811418197102333
Mediterranean	poc_mean_mg_m3	127	-2.401005239293518	132.24377180319308	110.11706567128557
Mediterranean	nflh_mean	127	-0.9173394308699583	0.053786370919911695	0.04980238563783633
Pacific	chl_mean_mg_m3	432	0.31676557085463775	2.752947391093162	0.9367923858851105
Pacific	poc_mean_mg_m3	432	0.410612232059557	365.9932019191848	97.34544476678137
Pacific	nflh_mean	432	0.6411661140130833	0.044837713190346344	0.032025390670095834
Southern	chl_mean_mg_m3	38	-3.795580814404878	1.2196556269910956	0.9523424074721988
Southern	poc_mean_mg_m3	38	-23.638212935360762	145.9942692288352	126.73067948999743
Southern	nflh_mean	38	0.007468383080701191	0.07413158982875905	0.04725445445441884
OVERALL	chl_mean_mg_m3	1151	0.5156353359806678	2.951822524448662	1.1808388864300026
OVERALL	poc_mean_mg_m3	1151	0.5317524547687325	289.0543084425235	110.69091265831484
OVERALL	nflh_mean	1151	0.5599985613761684	0.04920751095523455	0.033666900547231454