Update app.R

app.R

@@ -1090,35 +1090,44 @@ clean_college_data <- function(data, teams = NA) {
   data <- data %>% 
     filter(!(Level %in% c("D2", "JUCO", "NAIA", "D3")))
 
-  #predict xwoba
-  data <- data %>% mutate(row_id = row_number())
-
-  prob_cols <- c("prob_0", "prob_1", "prob_2", "prob_3", "prob_4")
-  
-  complete <- data %>%
-    filter(!is.na(ExitSpeed) & !is.na(Angle)) %>%
-    dplyr::select(-any_of(prob_cols))
-  
-  xwoba_preds <- predict(xwoba_model, complete, type = "prob")
-  colnames(xwoba_preds) <- prob_cols
-  
-  complete <- complete %>% bind_cols(xwoba_preds)
-  
-  data <- data %>%
-    dplyr::select(-any_of(prob_cols)) %>% 
-    left_join(complete %>% dplyr::select(row_id, any_of(prob_cols)), by = "row_id") %>%
-    dplyr::select(-row_id)
-  
-  data <- data %>% 
-    dplyr::select(-any_of(c("w1B", "w2B", "w3B", "wHR"))) %>% 
-    cross_join(college_guts %>% dplyr::select(w1B, w2B, w3B, wHR)) %>% 
-    mutate(xwOBA = w1B*prob_1 + w2B*prob_2 + w3B*prob_3 + wHR*prob_4 + 0*prob_0,
-           xwOBA = case_when(
-             KorBB == "Strikeout" ~ 0,
-             KorBB == "Walk" ~ .82,
-             event_type == "HBP" ~ .85,
-             TRUE ~ xwOBA
-           ))
+  #If able predict xwoba, if not enough data default to NA
+  data <- tryCatch({
+      d <- data %>% mutate(row_id = row_number())
+      prob_cols <- c("prob_0", "prob_1", "prob_2", "prob_3", "prob_4")
+  
+      complete <- d %>%
+        filter(!is.na(ExitSpeed) & !is.na(Angle)) %>%
+        dplyr::select(-any_of(prob_cols))
+  
+      xwoba_preds <- predict(xwoba_model, complete, type = "prob")
+      colnames(xwoba_preds) <- prob_cols
+  
+      complete <- complete %>% bind_cols(xwoba_preds)
+  
+      d <- d %>%
+        dplyr::select(-any_of(prob_cols)) %>%
+        left_join(complete %>% dplyr::select(row_id, any_of(prob_cols)), by = "row_id") %>%
+        dplyr::select(-row_id)
+  
+      d <- d %>%
+        dplyr::select(-any_of(c("w1B", "w2B", "w3B", "wHR"))) %>%
+        cross_join(college_guts %>% dplyr::select(w1B, w2B, w3B, wHR)) %>%
+        mutate(xwOBA = w1B*prob_1 + w2B*prob_2 + w3B*prob_3 + wHR*prob_4 + 0*prob_0,
+               xwOBA = case_when(
+                 KorBB == "Strikeout" ~ 0,
+                 KorBB == "Walk" ~ .82,
+                 event_type == "HBP" ~ .85,
+                 TRUE ~ xwOBA
+               ))
+      d
+    }, error = function(e) {
+      message("xwOBA prediction failed: ", conditionMessage(e), " \u2014 setting xwOBA and prob_* to NA.")
+      data %>% mutate(
+        prob_0 = NA_real_, prob_1 = NA_real_, prob_2 = NA_real_,
+        prob_3 = NA_real_, prob_4 = NA_real_,
+        xwOBA  = NA_real_
+      )
+    })
   
   message(nrow(data), " rows cleaned.")