Merge pull request #8 from pasmopy/yamashiro

Fix R script
pasmopy · Jun 19, 2021 · a5d5612 · a5d5612
2 parents c5d47be + 937331d
commit a5d5612
Showing 1 changed file with 2 additions and 22 deletions.
diff --git a/classification/brca_heatmap.R b/classification/brca_heatmap.R
@@ -58,28 +58,8 @@ annotation[is.na(annotation)] <- 0
 for (i in 1:nrow(annotation)){
   if (annotation$days_to_death[i] == 0){
     annotation$prognosis[i] <- 20
-  } else if (annotation$days_to_death[i] < 365 & annotation$days_to_death[i] > 0) {
-    annotation$prognosis[i] <- 1
-  } else if (annotation$days_to_death[i] > 366 & annotation$days_to_death[i] < 730){
-    annotation$prognosis[i] <- 2
-  } else if (annotation$days_to_death[i] > 731 & annotation$days_to_death[i] < 1095){
-    annotation$prognosis[i] <- 3
-  } else if (annotation$days_to_death[i] > 1096 & annotation$days_to_death[i] < 1825){
-    annotation$prognosis[i] <- 4
-  } else if (annotation$days_to_death[i] > 1826 & annotation$days_to_death[i] < 2190){
-    annotation$prognosis[i] <- 5
-  } else if (annotation$days_to_death[i] > 2191 & annotation$days_to_death[i] < 2555){
-    annotation$prognosis[i] <- 6
-  } else if (annotation$days_to_death[i] > 2556 & annotation$days_to_death[i] < 2920){
-    annotation$prognosis[i] <- 7
-  } else if (annotation$days_to_death[i] > 2921 & annotation$days_to_death[i] < 3650){
-    annotation$prognosis[i] <- 8
-  } else if (annotation$days_to_death[i] > 3651 & annotation$days_to_death[i] < 4015){
-    annotation$prognosis[i] <- 9
-  } else if (annotation$days_to_death[i] > 4016 & annotation$days_to_death[i] < 4380){
-    annotation$prognosis[i] <- 10
-  } else if (annotation$days_to_death[i] > 4381){
-    annotation$prognosis[i] <- 11
+  } else {
+    annotation$prognosis[i] <- floor(annotation$days_to_death[i]/365)
   }
 }