R/make_new_ranger_predictions.R

In jimtheflash/nba.modelR: What the Package Does (One Line, Title Case)

make_new_ranger_predictions <- function(use_case = 'team_games',
                                       ranger_model_objects, new_data_list) {

  if (use_case == 'team_games') {

    # loop through the new_data_list, select matching model, generate preds
    output_list <- list()
    for (nd in names(new_data_list)) {

      if (nrow(new_data_list[[nd]]) < 1) {
        message('no data for forecast for teams with ', nd, ' games played')
        next
      }

      ids <- new_data_list[[nd]] %>%
        dplyr::select(dplyr::ends_with('id')) %>%
        dplyr::mutate(range = nd,
                      type = 'ranger')

      predictors <- new_data_list[[nd]] %>%
        dplyr::select(-dplyr::ends_with('id'))

      model_list <- ranger_model_objects[[nd]][['model_list']]
      outcomes <- names(model_list)

      for (o in outcomes) {
browser()
        new_preds <- predict(model_list[[o]], predictors, type = 'response')$predictions

        if (inherits(new_preds, 'matrix') && ncol(new_preds) == 2) {
          new_preds <- as.numeric(new_preds[, "TRUE"])
        }

        ids[[o]] <- new_preds

      }

      output_list[[nd]] <- ids

    }

    output <- do.call(rbind, output_list)

    return(output)

  }

}