R/geometric_plot_compare.R

In bayesball/BayesTestStreak: Testing Streakiness in Binary Sequences by Bayes Factors

geometric_plot_compare <- function(spacings_df){
  
  spacings_df %>%
    group_by(BAT_ID, Spacing) %>% 
    summarize(N = n()) -> spacings_table
  
 ggplot(spacings_table, 
            aes(Spacing, log(N))) + 
            geom_point(size=3) +
    stat_smooth(method=lm, 
                se=FALSE, 
                formula = "y ~ x",
                linewidth=1, 
                color="black") +
    stat_smooth(method=loess, 
                se=FALSE, 
                formula = "y ~ x",
                linewidth=2) +
    facet_wrap(~ BAT_ID, nrow = 1) +
    theme_minimal() +
    theme(axis.text = element_text(size = rel(1.5))) +
    theme(axis.title = element_text(size = rel(1.5))) +
    xlab("Spacing") + 
    ylab("Log Frequency")
}