prdict.gam returned a 1D array instead of a vector

nflverse · Mar 1, 2021 · 7ace649 · 7ace649
1 parent e648e88
commit 7ace649
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/aggregate_game_stats.R b/R/aggregate_game_stats.R
@@ -526,7 +526,7 @@ add_dakota <- function(add_to_this, pbp, weekly) {
       dplyr::rename(player_id = .data$id) %>%
       dplyr::filter(.data$player_id %in% relevant_players)
 
-    model_data$dakota <- mgcv::predict.gam(dakota_model, model_data)
+    model_data$dakota <- mgcv::predict.gam(dakota_model, model_data) %>% as.vector()
 
     out <- add_to_this %>%
       dplyr::left_join(
@@ -547,7 +547,7 @@ add_dakota <- function(add_to_this, pbp, weekly) {
       dplyr::rename(player_id = .data$id) %>%
       dplyr::filter(.data$player_id %in% relevant_players)
 
-    model_data$dakota <- mgcv::predict.gam(dakota_model, model_data)
+    model_data$dakota <- mgcv::predict.gam(dakota_model, model_data) %>% as.vector()
 
     out <- add_to_this %>%
       dplyr::left_join(