File size: 1,618 Bytes
998922f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | #!/usr/bin/env Rscript
library(tidyverse)
library(plyr)
library(dplyr)
library(readr)
library(optparse)
save_fnorm_by_model <- function(df, factor) {
models <- unique(df$model)
for (mdl in models) {
df_by_model <- df |>
filter(model == mdl) |>
select(-c("model"))
sd <- sd(df_by_model$kurtosis_scaled)
mu <- mean(df_by_model$kurtosis_scaled)
df_by_model <- df_by_model |>
mutate(
kurtosis_scaled = ifelse(abs(kurtosis_scaled - mu) / sd > 3, factor, 1)
) |>
write_csv(paste0("fnorm-", mdl, ".csv"))
}
}
strip_name <- function(name) {
start <- nchar("fnorm-") + 1
stop <- nchar(name) - 4
return(substr(name, start, stop))
}
parser <- OptionParser()
parser <- add_option(
parser, c("-f", "--factor"),
type = "double",
help = "Factor to apply",
metavar = "double"
)
parser <- add_option(
parser, c("-d", "--data_dir"),
type = "character",
help = "Data directory of fnorm csv files",
metavar = "character"
)
args <- parse_args(parser)
if (is.null(args$data_dir)) {
fnorm_dir <- "../src/data"
} else {
fnorm_dir <- args$data_dir
}
if (is.null(args$factor)) {
factor <- 2.0
} else {
factor <- args$factor
}
fnorm_dir <- "/tmp/apportion-sensi"
fnorm_dir <- path.expand(fnorm_dir)
fnorm_fps <- dir(
path = fnorm_dir,
pattern = "fnorm-.*\\.csv$",
full.names = TRUE
)
names(fnorm_fps) <- sapply((basename(fnorm_fps)), strip_name)
df_fnorm <- plyr::ldply(
fnorm_fps,
read.csv,
stringsAsFactors = FALSE,
.id = "model"
)
# df_fnorm <- df_fnorm |>
# select(!c("kurtosis_scaled"))
save_fnorm_by_model(df_fnorm, factor)
|