Can save and plot with/out labels and phenotypes

AJResearchGroup · May 18, 2022 · 0b91f63 · 0b91f63
1 parent 8d433b8
commit 0b91f63
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 18 deletions.
diff --git a/R/create_plots_from_gcae_experiment_results.R b/R/create_plots_from_gcae_experiment_results.R
@@ -49,11 +49,15 @@ create_plots_from_gcae_experiment_results <- function( # nolint indeed a long fu
       csv_filenames$genotype_concordances_filename,
     png_filename = png_filenames$genotype_concordances_filename
   )
-  gcaer::plot_phenotype_predictions_from_file(
-    phenotype_predictions_filename =
-      csv_filenames$phenotype_predictions_filename,
-    png_filename = png_filenames$phenotype_predictions_filename
-  )
+  if (file.exists(csv_filenames$phenotype_predictions_filename)) {
+    gcaer::plot_phenotype_predictions_from_file(
+      phenotype_predictions_filename =
+        csv_filenames$phenotype_predictions_filename,
+      png_filename = png_filenames$phenotype_predictions_filename
+    )
+  } else {
+    png_filenames$phenotype_predictions_filename <- NULL
+  }
   gcaer::plot_train_times_from_file(
     train_times_filename = csv_filenames$train_times_filename,
     png_filename = png_filenames$train_times_filename
@@ -66,9 +70,13 @@ create_plots_from_gcae_experiment_results <- function( # nolint indeed a long fu
     losses_from_train_v_filename = csv_filenames$losses_from_train_v_filename,
     png_filename = png_filenames$losses_from_train_v_filename
   )
-  gcaer::plot_nmse_in_time_from_file(
-    nmse_in_time_filename = csv_filenames$nmse_in_time_filename,
-    png_filename = png_filenames$nmse_in_time_filename
-  )
+  if (file.exists(csv_filenames$nmse_in_time_filename)) {
+    gcaer::plot_nmse_in_time_from_file(
+      nmse_in_time_filename = csv_filenames$nmse_in_time_filename,
+      png_filename = png_filenames$nmse_in_time_filename
+    )
+  } else {
+    png_filenames$nmse_in_time_filename <- NULL
+  }
   png_filenames
 }
diff --git a/R/save_gcae_experiment_results.R b/R/save_gcae_experiment_results.R
@@ -26,6 +26,7 @@ save_gcae_experiment_results <- function(
   gcaer::check_folder_name(folder_name)
 
   filenames <- gcaer::get_gcae_experiment_results_filenames(folder_name)
+
   dir.create(folder_name, showWarnings = FALSE, recursive = TRUE)
   readr::write_csv(
     x = gcae_experiment_results$score_per_pop_table,
@@ -39,10 +40,14 @@ save_gcae_experiment_results <- function(
     gcae_experiment_results$genotype_concordances_table,
     file = filenames$genotype_concordances_filename
   )
-  readr::write_csv(
-    gcae_experiment_results$phenotype_predictions_table,
-    file = filenames$phenotype_predictions_filename
-  )
+  if ("phenotype_predictions_table" %in% names(gcae_experiment_results)) {
+    readr::write_csv(
+      gcae_experiment_results$phenotype_predictions_table,
+      file = filenames$phenotype_predictions_filename
+    )
+  } else {
+    filenames$phenotype_predictions_filename <- NULL
+  }
   readr::write_csv(
     gcae_experiment_results$train_times_table,
     file = filenames$train_times_filename
@@ -55,10 +60,13 @@ save_gcae_experiment_results <- function(
     gcae_experiment_results$losses_from_train_v_table,
     file = filenames$losses_from_train_v_filename
   )
-  readr::write_csv(
-    gcae_experiment_results$nmse_in_time_table,
-    file = filenames$nmse_in_time_filename
-  )
-
+  if ("nmse_in_time_table" %in% names(gcae_experiment_results)) {
+    readr::write_csv(
+      gcae_experiment_results$nmse_in_time_table,
+      file = filenames$nmse_in_time_filename
+    )
+  } else {
+    filenames$nmse_in_time_filename <- NULL
+  }
   filenames
 }