FuzzyNaiveBayes
Fuzzy Naive Bayes
Examples
set.seed(1) # determining a seed
data(HouseVotes84)
# Splitting into Training and Testing
split <- caTools::sample.split(t(HouseVotes84[, 1]), SplitRatio = 0.7)
Train <- subset(HouseVotes84, split == "TRUE")
Test <- subset(HouseVotes84, split == "FALSE")
# ----------------
# matrix or data frame of test set cases.
# A vector will be interpreted as a row vector for a single case.
test <- Test[, -1]
fit_FNB <- FuzzyNaiveBayes(
train = Train[, -1],
cl = Train[, 1]
)
pred_FNB <- predict(fit_FNB, test)
head(pred_FNB)
#> [1] republican democrat republican republican republican democrat
#> Levels: democrat republican
head(Test[, 1])
#> [1] democrat democrat democrat republican republican democrat
#> Levels: democrat republican