-
Notifications
You must be signed in to change notification settings - Fork 0
/
Naive Bayes.r
30 lines (17 loc) · 826 Bytes
/
Naive Bayes.r
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
## Naive Bayes ##
# Importando os dados
library(xlsx)
planejamento <- readxl::read_xlsx("C:\\Users\\lopes\\Dropbox\\Introducao a Mineracao de Dados\\Capítulo 3 - Análise Preditiva\\3.1 - Classificacao\\Naive Bayes.xlsx")
head(planejamento)
exemplar_teste <- data.frame(PREVISAO="sol", TEMPERATURA="frio", HUMIDADE="normal", VENTO="sim")
# Carregando pacote que implementa o Naive Bayes
library(e1071)
?naiveBayes # para construir o classificador ("modelo")
?predict.naiveBayes # implementa a predição usando o Naive Bayes (pode-se usar somente a função predict() )
# Criando o modelo
modelo_NB <- naiveBayes(planejamento[1:4], planejamento$VENDA)
modelo_NB
# Implementando a classificação utilizando a função predict():
# type="raw" e sem correção de Laplace:
p <- predict(modelo_NB, exemplar_teste, type="raw")
p