mirror of
https://github.com/andre-wojtowicz/uci-ml-to-r.git
synced 2024-11-03 14:20:28 +01:00
30 lines
1012 B
R
30 lines
1012 B
R
|
preprocessDataset = function()
|
||
|
|
{
|
||
|
|
#set.seed(SEED)
|
||
|
|
|
||
|
|
arff.file = "seismic-bumps.arff"
|
||
|
|
|
||
|
|
dataset = read.arff(paste0(orig.dir, "/", arff.file))
|
||
|
|
|
||
|
|
dataset = dataset %>% select(-c(nbumps6:nbumps89)) %>%
|
||
|
|
mutate(genergy=as.integer(genergy),
|
||
|
|
gpuls=as.integer(gpuls),
|
||
|
|
gdenergy=as.integer(gdenergy),
|
||
|
|
gdpuls=as.integer(gdpuls),
|
||
|
|
nbumps=as.integer(nbumps),
|
||
|
|
nbumps2=as.integer(nbumps2),
|
||
|
|
nbumps3=as.integer(nbumps3),
|
||
|
|
nbumps4=as.integer(nbumps4),
|
||
|
|
nbumps5=as.integer(nbumps5),
|
||
|
|
energy=as.integer(energy),
|
||
|
|
maxenergy=as.integer(maxenergy)
|
||
|
|
)
|
||
|
|
|
||
|
|
#dataset.1 = dataset %>% filter(class == "1")
|
||
|
|
#dataset.0 = dataset %>% filter(class == "0") %>%
|
||
|
|
# sample_n(nrow(dataset.1)*4)
|
||
|
|
#
|
||
|
|
#dataset = rbind(dataset.0, dataset.1)
|
||
|
|
|
||
|
|
return(dataset)
|
||
|
|
}
|