compute ms ssim in test steps

broadinstitute · Jan 28, 2025 · ef1de28 · ef1de28
1 parent 8461b7f
commit ef1de28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml4h/models/train.py b/ml4h/models/train.py
@@ -278,7 +278,7 @@ def interpolate_controlled_generations(diffuser, tensor_maps_out, control_tm, ba
                     plt.plot(generated_images[i_col, :, lead], label=lead)
             elif len(generated_images.shape) == 4:
                 plt.imshow(generated_images[i_col], cmap='gray')
-            plt.gca().set_title(f'{control_tm.name[:10]}: {pheno_scale:0.1f}')
+            #plt.gca().set_title(f'{control_tm.name[:10]}: {pheno_scale:0.1f}')
             plt.axis("off")
 
     plt.tight_layout()