R/plot_cluster_means.R

In danichusfu/RouteIdentification: Route and cluster assignment for NFL tracking data from NextGenStats

#' plot cluster means given the results of the em
#'
#' @param em_results results of the em algorithm
#' @importFrom magrittr %>%
#' @return plot of the cluster means

plot_cluster_means <- function(em_results){
  Beta <- em_results$Beta
  tibble::tibble(Beta = Beta) %>%
    dplyr::mutate(mean_curve = purrr::map(Beta, ~ bezier::bezier(t = seq(0, 1, by = 0.1), p = .) %>% tibble::as_tibble()),
                  cluster = dplyr::row_number()) %>%
    dplyr::select(-Beta) %>%
    tidyr::unnest() %>%
    ggplot2::ggplot(aes(x = V1, y = V2, colour = factor(cluster))) +
    ggplot2::geom_path() +
    ggplot2::theme_bw() 
}