#if(!require(devtools)) install.packages("devtools")
#devtools::install_github("naoki-egami/dsl", dependencies = TRUE)
library(dsl)

power_csv <-"~/dsl/093025_power_dsl.csv"
power_df <- read.csv(power_csv, header = TRUE) 

power_c1 <- power_df |>
  filter(source=='c1') 

power_c2 <- power_df |>
  filter(source=='c2') 

power_c3 <- power_df |>
  filter(source=='c3') 

power_model <- power_dsl(
  labeled_size = c(100, 200, 300, 600, 1000),
  model = "logit", 
  formula = dsl_score ~ human_BE_prop + 
    median_gerrit_delta + median_gerrit_reviewers + 
    n_comments + 
    priority_score +
    week_index,
  predicted_var = "human_BE_prop",
  prediction = "olmo_BE_prop",
  data=power_c1
)

summary(power_model)
plot(power_model, coef_name = "human_BE_prop")

trial_model <- dsl(
  model = "logit", 
  formula = dsl_score ~ human_BE_prop + 
    median_gerrit_delta + median_gerrit_reviewers + 
    n_comments + 
    priority_score +
    week_index,
  predicted_var = "human_BE_prop",
  prediction = "olmo_BE_prop",
  data=power_df
)
summary(trial_model)