mirror of
https://github.com/andre-wojtowicz/uci-ml-to-r.git
synced 2024-07-22 07:35:30 +02:00
24 lines
768 B
R
24 lines
768 B
R
preprocess.dataset = function()
|
|
{
|
|
csv.file.w = "winequality-white.csv"
|
|
csv.file.r = "winequality-red.csv"
|
|
|
|
dataset.w = read.csv(file.path(orig.dir, "/", csv.file.w), sep = ";",
|
|
check.names = FALSE)
|
|
dataset.w = dataset.w %>%
|
|
mutate(color = "white")
|
|
|
|
dataset.r = read.csv(paste0(orig.dir, "/", csv.file.r), sep = ";",
|
|
check.names = FALSE)
|
|
dataset.r = dataset.r %>%
|
|
mutate(color = "red")
|
|
|
|
dataset =
|
|
rbind(dataset.w, dataset.r) %>%
|
|
mutate(color = factor(color),
|
|
quality = ifelse(quality > 5, 1, 0)) %>%
|
|
select(`fixed acidity`:alcohol, color, quality) %>%
|
|
mutate(quality = factor(quality))
|
|
|
|
return(dataset)
|
|
} |