Upload folder from local machine

.gitattributes

@@ -67,3 +67,31 @@ data/processed/scPerturb/latent_flow_replogle_k562_essential/figures/08_latent_s
 data/processed/scPerturb/latent_flow_replogle_k562_essential/figures/11_split_gene_count_heatmap_subset.png filter=lfs diff=lfs merge=lfs -text
 data/processed/scPerturb/latent_flow_replogle_k562_essential/replogle_k562_essential_latent_flow_ready.h5ad filter=lfs diff=lfs merge=lfs -text
 data/processed/scPerturb/prediction_ready/replogle_k562_essential_single_gene_prediction_ready_full.h5ad filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_A_gene_mean_calibration_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_B_MA_plot_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_DE_heatmap_delta_predicted_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_DE_heatmap_delta_truth_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_F_perturb_corr_truth_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_G_perturb_corr_clustered_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_H_cell_PCA_truth_pred_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_J_metric_panels_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/bio_K_topvar_gene_scatters_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/evaluation_metrics_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/figures/training_history_seen_cell_split_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/pred_conditional_flow_matching_method_seen_cell_split_test.h5ad filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/conditional_flow_matching_method/real_conditional_flow_matching_method_seen_cell_split_test.h5ad filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/foundation_encoder_flow_matching/pred_foundation_encoder_flow_matching_seen_cell_split_test.h5ad filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/foundation_encoder_flow_matching/real_foundation_encoder_flow_matching_seen_cell_split_test.h5ad filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_A_gene_mean_calibration_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_DE_heatmap_delta_predicted_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_DE_heatmap_delta_truth_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_F_perturb_corr_truth_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_G_perturb_corr_clustered_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_H_cell_PCA_truth_pred_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_J_metric_panels_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_K_topvar_gene_scatters_use_existing_test.pdf filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/bio_K_topvar_gene_scatters_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/evaluation_metrics_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/figures/training_history_use_existing_test.png filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/pred_gears_from_scratch_use_existing_test.h5ad filter=lfs diff=lfs merge=lfs -text
+baseline/outputs/gears_from_scratch/real_gears_from_scratch_use_existing_test.h5ad filter=lfs diff=lfs merge=lfs -text

baseline/outputs/conditional_flow_matching_method/aggregate_conditional_flow_matching_method_seen_cell_split_test.csv

@@ -0,0 +1,7 @@
+,mean,median
+n_cells,14.574416342412452,12.0
+mse_mean,0.01917405563267582,0.013619688805192709
+mae_mean,0.09751567736916708,0.08807016909122467
+pearson_mean,0.9756459682686928,0.9822781238831229
+pearson_delta,0.08749359142595428,0.07193818371390132
+delta_l2,8.496514736910274,7.682950496673584

baseline/outputs/conditional_flow_matching_method/conditional_flow_matching_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb5d5bd43d02a10f37f5829c074eccd53260d67a11b828cbb54eb0f00b64eda4
+size 68058135

baseline/outputs/conditional_flow_matching_method/config_conditional_flow_matching_method_seen_cell_split_test.json

@@ -0,0 +1,45 @@
+{
+  "name": "conditional_flow_matching_method",
+  "input_path": "/root/final/data/processed/scPerturb/prediction_ready/replogle_k562_essential_single_gene_prediction_ready_full.h5ad",
+  "split_strategy": "seen_cell_split",
+  "train_frac": 0.8,
+  "val_frac": 0.1,
+  "eval_split": "test",
+  "random_seed": 42,
+  "latent_dim": 128,
+  "hidden_dim": 512,
+  "condition_dim": 256,
+  "time_dim": 128,
+  "dropout": 0.08,
+  "num_flow_blocks": 6,
+  "batch_size": 128,
+  "epochs": 20,
+  "steps_per_epoch": 200,
+  "validation_steps": 48,
+  "learning_rate": 0.001,
+  "lr_scheduler": "plateau",
+  "plateau_factor": 0.5,
+  "plateau_patience": 3,
+  "early_stop_patience": 5,
+  "fixed_val_subset": true,
+  "flow_lr_mult": 1.5,
+  "uncertainty_samples": 1,
+  "weight_decay": 1e-05,
+  "recon_weight": 1.0,
+  "flow_weight": 1.5,
+  "endpoint_weight": 0.5,
+  "mmd_weight": 0.1,
+  "mean_weight": 1.0,
+  "noise_std": 0.05,
+  "sinkhorn_epsilon": 0.5,
+  "sinkhorn_iters": 30,
+  "inference_steps": 24,
+  "n_train_conditions": 2056,
+  "n_eval_conditions": 2056,
+  "n_unseen_eval_conditions": 0,
+  "feature_gene_columns": [
+    "is_feature_gene",
+    "highly_variable"
+  ],
+  "model_path": "/root/final/baseline/outputs/conditional_flow_matching_method/conditional_flow_matching_model.pt"
+}

baseline/outputs/conditional_flow_matching_method/pred_conditional_flow_matching_method_seen_cell_split_test.h5ad

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4a12ccdafff9ea74f69da33f744fb0c7529d8c7cc0354c5aa6568a38113fc4b
+size 396168804

baseline/outputs/conditional_flow_matching_method/real_conditional_flow_matching_method_seen_cell_split_test.h5ad

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f435139d91b9f7994acdfa8b7fb25b0e476e890384fb010f3a32d1ff55b77de
+size 136056964

baseline/outputs/conditional_flow_matching_method/summary_conditional_flow_matching_method_seen_cell_split_test.csv

@@ -0,0 +1,16 @@
+,value
+method_name,conditional_flow_matching_method
+input_path,/root/final/data/processed/scPerturb/prediction_ready/replogle_k562_essential_single_gene_prediction_ready_full.h5ad
+eval_split,test
+train_split,train
+val_split,val
+n_control_cells_train,8653
+n_eval_cells,29965
+n_train_genes,2056
+n_eval_genes,2056
+n_eval_genes_missing_from_train,0
+mse_mean_avg,0.01917405563267582
+mae_mean_avg,0.09751567736916708
+pearson_mean_avg,0.9756459682686928
+pearson_delta_avg,0.08749359142595428
+delta_l2_avg,8.496514736910274

baseline/outputs/conditional_flow_matching_method/training_history_conditional_flow_matching_method_seen_cell_split_test.csv

@@ -0,0 +1,20 @@
+epoch,train_total,train_recon,train_flow,train_endpoint,train_mmd,train_mean,val_total,val_recon,val_flow,val_endpoint,val_mmd,val_mean
+1,0.7229754361510277,0.28971150815486907,0.24020878426264972,0.0787694172328338,0.19176664233207702,0.014389381218934431,0.33742387033998966,0.2706386561815937,0.015656325869106997,0.006038070787326433,0.23440208037694296,0.01684147972264327
+2,0.3107110871374607,0.27675333626568316,0.00535464484593831,0.002205554904357996,0.182200910449028,0.00660291428968776,0.308100796615084,0.26800615433603525,0.0006917850102278559,0.0006494638643440945,0.23177250971396765,0.015554978871174777
+3,0.2989280115067959,0.2729910775274038,0.000976635105907917,0.0007381650074967184,0.18184845447540282,0.005918052765191533,0.3080284483730793,0.2683586037407319,0.0002501830507147436,0.0005082318154260671,0.23215054720640182,0.01582539805288737
+4,0.29797442868351937,0.27295888148248193,0.0005482676622341387,0.00059736281473306,0.18103294491767882,0.005791169993462972,0.3059702701866627,0.2671983763575554,0.000187529703907785,0.0004899104587821057,0.23031875739494959,0.015213765189400874
+5,0.30014777824282646,0.27508261904120446,0.00039250255038496105,0.0005459327067364938,0.1818482404947281,0.0060186157783027735,0.30635998025536537,0.26757268980145454,0.00014019711034052307,0.0004674413861115075,0.23050070305665335,0.015293202945031226
+6,0.3017364460229874,0.27629068948328495,0.00032005466979171613,0.0005194408107490745,0.18273981511592866,0.006431970063131303,0.3054399124036233,0.2670242730528116,8.965702742595265e-05,0.0004489487655519042,0.22983028491338095,0.01507365114230197
+7,0.2976172469556332,0.2728036333620548,0.0002161072821036214,0.00048459474943228996,0.18328052818775176,0.005919103824999184,0.3058415297418833,0.2673094707230727,7.187519903103141e-05,0.00043725281102524605,0.22969928135474524,0.015235687426563041
+8,0.296335127428174,0.27198477134108545,0.00017527998832520098,0.0004695315036224201,0.1817716372013092,0.005675507162231952,0.3069052553425233,0.2680856008082628,3.851303586088761e-05,0.0004033020256125989,0.230217474202315,0.015538490028120577
+9,0.29768055096268653,0.27330787897109987,0.0001349272867446416,0.0004576769690902438,0.18224453866481782,0.005716988170170225,0.30487822989622754,0.2667280314490199,4.05872543372728e-05,0.0004374973335264561,0.22910139709711075,0.014960426201772256
+10,0.3011392876505852,0.2761539756506681,0.00010996463399351342,0.0004482559330062941,0.18272344827651976,0.006323891071369872,0.3066509397079547,0.26794911175966263,2.592924249711359e-05,0.00041573257476557046,0.2301128382484118,0.015443787609304612
+11,0.29784899413585664,0.2736397182941437,0.00010055587150418432,0.0004440271585190203,0.18025926172733306,0.005810505092376843,0.30489887048800784,0.2667912198230624,2.4591204957384132e-05,0.0004189171310523913,0.2285609394311905,0.01500520906605137
+12,0.2998918299376965,0.27562876991927626,8.157462834788021e-05,0.0004382953247113619,0.1807991510629654,0.00584163500694558,0.30497980924944085,0.26691776514053345,3.130836391089057e-05,0.000427247950938181,0.22808108975489935,0.014993348711868748
+13,0.29903030902147293,0.2748288442194462,6.922077820490813e-05,0.00043506107467692343,0.18069200038909913,0.005810903518577107,0.3049966997156541,0.26690145892401534,1.7661998204706226e-05,0.00042133735526779975,0.22828781604766846,0.015029298442338282
+14,0.29681979984045026,0.2731965658068657,4.984963290553424e-05,0.00042991104259272107,0.180612171292305,0.005272286800318398,0.304299861813585,0.26653613056987524,6.265737804748521e-06,0.00041442390041387017,0.22698897620042166,0.01484821921621915
+15,0.29439964607357977,0.27132133327424524,4.378180017738487e-05,0.0004238814547716174,0.1789425027370453,0.004906447771354579,0.3053674524029096,0.2672811135028799,5.2374380269763305e-06,0.00039320610176218906,0.22708587100108465,0.015173293242696673
+16,0.298062794059515,0.2745291447639465,4.0351100269617743e-05,0.00042377187666716056,0.17960356652736664,0.00530088092375081,0.3044471361984809,0.26659432239830494,4.9299719648843165e-06,0.0004348547772679012,0.22777657707532248,0.014850332687880533
+17,0.2954445379972458,0.2721785414963961,3.7617002762999616e-05,0.0004223160068795551,0.17852332472801208,0.0051460796699393544,0.3047699636469285,0.2668332125370701,5.056588274025368e-06,0.00043147314681846183,0.22774638483921686,0.01493879126792308
+18,0.2971596910059452,0.2734533756971359,3.491570643745945e-05,0.000420169604185503,0.17951957285404205,0.005491900352062658,0.30458676256239414,0.26674242317676544,5.0952250821486205e-06,0.00042136689626204316,0.22738620887200037,0.014887394999580769
+19,0.29455566182732584,0.2715554262697697,3.188204713296727e-05,0.0004176008127979003,0.17817343950271605,0.004926269159186631,0.30446241423487663,0.26676911767572165,3.317419592955654e-06,0.0003996163850388257,0.2258559688925743,0.014902912473189645

baseline/outputs/foundation_encoder_flow_matching/aggregate_foundation_encoder_flow_matching_seen_cell_split_test.csv

@@ -0,0 +1,7 @@
+,mean,median
+n_cells,14.574416342412452,12.0
+mse_mean,0.01941383839131909,0.013922772370278835
+mae_mean,0.09817201291403012,0.08879996463656425
+pearson_mean,0.9754807498099398,0.9819993528020536
+pearson_delta,0.09023290415007162,0.06830840871113207
+delta_l2,8.575719317573517,7.767965793609619

baseline/outputs/foundation_encoder_flow_matching/config_foundation_encoder_flow_matching_seen_cell_split_test.json

@@ -0,0 +1,48 @@
+{
+  "name": "foundation_encoder_flow_matching",
+  "input_path": "/root/final/data/processed/scPerturb/prediction_ready/replogle_k562_essential_single_gene_prediction_ready_full.h5ad",
+  "split_strategy": "seen_cell_split",
+  "train_frac": 0.8,
+  "val_frac": 0.1,
+  "eval_split": "test",
+  "random_seed": 42,
+  "foundation_model_name": "geneformer",
+  "foundation_embed_source": "hvg_only",
+  "foundation_obsm_key": "X_geneformer",
+  "foundation_embed_h5ad_path": null,
+  "foundation_embed_file": null,
+  "foundation_obs_csv": null,
+  "use_dual_channel": true,
+  "foundation_freeze": true,
+  "foundation_proj_dim": 128,
+  "hvg_proj_dim": 64,
+  "latent_dim": 96,
+  "hidden_dim": 256,
+  "condition_dim": 128,
+  "time_dim": 64,
+  "dropout": 0.1,
+  "num_flow_blocks": 4,
+  "batch_size": 128,
+  "epochs": 20,
+  "steps_per_epoch": 200,
+  "validation_steps": 32,
+  "learning_rate": 0.001,
+  "weight_decay": 1e-05,
+  "recon_weight": 1.0,
+  "flow_weight": 1.0,
+  "endpoint_weight": 0.5,
+  "mmd_weight": 0.1,
+  "mean_weight": 1.0,
+  "noise_std": 0.05,
+  "sinkhorn_epsilon": 0.5,
+  "sinkhorn_iters": 30,
+  "inference_steps": 24,
+  "n_train_conditions": 2056,
+  "n_eval_conditions": 2056,
+  "n_unseen_eval_conditions": 0,
+  "feature_gene_columns": [
+    "is_feature_gene",
+    "highly_variable"
+  ],
+  "model_path": "/root/final/baseline/outputs/foundation_encoder_flow_matching/foundation_encoder_flow_matching_model.pt"
+}

baseline/outputs/foundation_encoder_flow_matching/foundation_encoder_flow_matching_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:72b9c6fe14d958a38108987e2a32fe72bdfb52c90c972f282541afc8e66c2289
+size 24385525

baseline/outputs/foundation_encoder_flow_matching/pred_foundation_encoder_flow_matching_seen_cell_split_test.h5ad

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f56da7b352ecc2474992ddc8a42fd0f74fa7541955c9b723a215fe29bf38bb0
+size 394862904

baseline/outputs/foundation_encoder_flow_matching/real_foundation_encoder_flow_matching_seen_cell_split_test.h5ad

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff105a82e83979e01c742ac0466c814c11930c17411a8fc4ec16590d67eebbfb
+size 136058292

baseline/outputs/foundation_encoder_flow_matching/training_history_foundation_encoder_flow_matching_seen_cell_split_test.csv

@@ -0,0 +1,21 @@
+epoch,train_total,train_recon,train_flow,train_endpoint,train_mmd,train_mean,val_total,val_recon,val_flow,val_endpoint,val_mmd,val_mean
+1,0.38691789776086805,0.29517462618649004,0.04904787425184622,0.016468857816653325,0.18808757424354552,0.015652211549459026,0.3274608151987195,0.275825597345829,0.001044323875248665,0.0007793883223712328,0.24831904470920563,0.02536929867346771
+2,0.30455235108733175,0.2775276107341051,0.0023269738734234126,0.0011926707287784667,0.177983837723732,0.006303047815454193,0.3212786726653576,0.27292283019050956,0.0004433240710568498,0.0005435769221548981,0.2625858522951603,0.021382144579547457
+3,0.29915610790252684,0.2738037461787462,0.0012439033997361547,0.0008200335814035497,0.17889966309070587,0.005808474900550209,0.32495994027704,0.27975515043362975,0.0003343683379171125,0.0004975125066266628,0.24732991307973862,0.01988867791078519
+4,0.29836430445313455,0.27362029798328874,0.0007742353723733685,0.0006675001711118966,0.1789140635728836,0.005744616611627862,0.3281898917630315,0.2743259877897799,0.00012644549792639737,0.0004485846257011872,0.2576749287545681,0.027745677973143756
+5,0.30053194269537925,0.27569156885147095,0.0005305912837502547,0.0005861266049032566,0.18015105187892913,0.006001613531843759,0.31706626154482365,0.271136119030416,8.207476935240265e-05,0.00044235233008294017,0.25808728113770485,0.019818157656118274
+6,0.3020606791973114,0.2767776419222355,0.00045562463070382365,0.0005597601740737445,0.1812485247850418,0.006422679775278084,0.3101926790550351,0.26925651589408517,6.836948966792988e-05,0.000447841421191697,0.2147996462881565,0.019163907432812266
+7,0.29798629999160764,0.27322004564106467,0.00040472575463354585,0.0005387418481404893,0.18194822669029237,0.005897335183690302,0.3116888366639614,0.27143998397514224,5.981437584523519e-05,0.0004106282340217149,0.2355927899479866,0.0164244404877536
+8,0.296607558131218,0.2723300854116678,0.00031016745124361477,0.0005022020945034455,0.18060479462146758,0.005655723655945621,0.3245162405073643,0.2692126724869013,6.881837271066615e-05,0.0004348013458184141,0.22198719158768654,0.032818628882523626
+9,0.2979561977088451,0.2736369680613279,0.00024457521263684614,0.00048595993110211566,0.18126410663127898,0.005705264492426068,0.3120060069486499,0.2671913858503103,0.00015860154053370934,0.00043401215498306556,0.2274879813194275,0.02169021778536262
+10,0.30144763812422753,0.27645000398159025,0.000256421913363738,0.0004827430541627109,0.18179213345050813,0.006320627338718623,0.32949820067733526,0.27811660477891564,7.695834995047335e-05,0.0004271000398148317,0.2738311104476452,0.0237079802172957
+11,0.2980618551373482,0.27387773282825945,0.0002043454532758915,0.0004651212583121378,0.17950365245342254,0.005796851214254275,0.31166065484285355,0.2688261936418712,4.122563217379138e-05,0.00043036165152443573,0.24388854950666428,0.018189195747254416
+12,0.3000770792365074,0.2758459587395191,0.00016575749519688544,0.000456608561653411,0.18008724153041838,0.0058283342176582665,0.30728677567094564,0.265851276461035,3.8194732212559757e-05,0.00042105488319066353,0.23691591247916222,0.01749518767610425
+13,0.2992090843617916,0.275026483386755,0.0001538951452312176,0.00045235942336148584,0.18003856003284455,0.005798668235074729,0.3169968528673053,0.26816306728869677,5.754179380801361e-05,0.0004298104349800269,0.2489001788198948,0.02367131452774629
+14,0.29762876868247984,0.2737561973184347,0.00015962561763444683,0.000448978004133096,0.1805049878358841,0.005437960230046883,0.3246108265593648,0.27808606857433915,3.6118976822763216e-05,0.00040605286949357833,0.22163715586066246,0.02412189680035226
+15,0.2950549632310867,0.27182348765432834,0.00012080478607458645,0.00044040873035555705,0.17867071449756622,0.005023393639712595,0.303118328563869,0.26511960616335273,3.0777167353335244e-05,0.0004082728028151905,0.2235776223242283,0.015406048740260303
+16,0.29867891550064085,0.2749717626720667,0.00010667893966456176,0.0004333837666490581,0.17947034358978273,0.005436747506028041,0.31389718502759933,0.27396360924467444,3.137197688829474e-05,0.00040794723190629156,0.22455397248268127,0.01724283956355066
+17,0.2960674087703228,0.2726463632285595,9.388849945025868e-05,0.0004296518721093889,0.17837208151817321,0.005275122465100139,0.32834118511527777,0.2750188857316971,1.9406186737569442e-05,0.0004097648852621205,0.2599041238427162,0.02710759741603397
+18,0.2977588853240013,0.27388852305710315,9.371036765514873e-05,0.0004268587040132843,0.17940838932991027,0.005622384629095905,0.31483375933021307,0.2713820734061301,8.290409971323243e-05,0.00043064571400464047,0.2412523627281189,0.01902822239208035
+19,0.29542832225561144,0.27215037629008293,9.678928849098156e-05,0.00042518379297689534,0.17828599512577056,0.005139965713606216,0.31950006540864706,0.2781221577897668,7.059473591652932e-05,0.00044125305976194795,0.2129979096353054,0.01978689779934939
+20,0.2930982676148415,0.2697341851145029,8.121039700199617e-05,0.00042242487950716167,0.17926863789558412,0.005144796972745098,0.3244615225121379,0.28195606358349323,2.0344247758430356e-05,0.00040426405666948995,0.23186839371919632,0.019096140211331658

baseline/outputs/gears_from_scratch/aggregate_gears_from_scratch_use_existing_test.csv

@@ -0,0 +1,7 @@
+,mean,median
+n_cells,164.25405405405405,141.0
+mse_mean,0.010499478619847749,0.009012488648295403
+mae_mean,0.06984614715785593,0.0652008205652237
+pearson_mean,0.9866680353374947,0.989098821996637
+pearson_delta,0.1809813825799865,0.16880085186328223
+delta_l2,6.636925668974181,6.249810218811035

baseline/outputs/gears_from_scratch/config_gears_from_scratch_use_existing_test.json

@@ -0,0 +1,28 @@
+{
+  "name": "gears_from_scratch",
+  "input_path": "/root/final/data/processed/scPerturb/prediction_ready/replogle_k562_essential_single_gene_prediction_ready_full.h5ad",
+  "split_strategy": "use_existing",
+  "eval_split": "test",
+  "random_seed": 42,
+  "graph_topk": 20,
+  "max_graph_control_cells": 1500,
+  "top_de_genes": 50,
+  "hidden_dim": 64,
+  "num_gnn_layers": 3,
+  "dropout": 0.1,
+  "batch_size": 32,
+  "epochs": 20,
+  "steps_per_epoch": 200,
+  "validation_steps": 40,
+  "learning_rate": 0.001,
+  "weight_decay": 1e-05,
+  "de_mse_lambda": 1.0,
+  "direction_lambda": 0.5,
+  "n_genes": 4334,
+  "n_train_conditions_all": 1645,
+  "n_train_conditions_graph": 1494,
+  "n_eval_conditions_all": 205,
+  "n_eval_conditions_graph": 185,
+  "n_eval_conditions_missing_graph": 20,
+  "model_path": "/root/final/baseline/outputs/gears_from_scratch/gears_from_scratch_model.pt"
+}