Boa noite Pessoal,
Quando eu executo o programa, eu recebo seguinte mensagem de erro:
Traceback (most recent call last):
File "classificacao.py", line 22, in <module>
resultado = modelo.predict(misterioso)
File "C:\Python27amd64\lib\site-packages\sklearn\naive_bayes.py", line 66, in predict
jll = self._joint_log_likelihood(X)
File "C:\Python27amd64\lib\site-packages\sklearn\naive_bayes.py", line 724, in _joint_log_likelihood
X = check_array(X, accept_sparse='csr')
File "C:\Python27amd64\lib\site-packages\sklearn\utils\validation.py", line 441, in check_array
"if it contains a single sample.".format(array))
ValueError: Expected 2D array, got 1D array instead:
array=[1 1 1].
Reshape your data either using array.reshape(-1, 1) if your data has a single feature or array.reshape(1, -1) if it contains a single sample.
Segue o código que estou executando:
# eh gordinho? tem perninha curta? faz au au ?
porco1 = [1,1,0]
porco2 = [1,1,0]
porco3 = [1,1,0]
cachorro1 = [1,1,1]
cachorro2 = [0,1,1]
cachorro3 = [0,1,1]
dados = [porco1, porco2, porco3, cachorro1, cachorro2, cachorro3]
marcacoes = [1, 1, 1,-1,-1,-1]
misterioso = [1,1,1]
from sklearn.naive_bayes import MultinomialNB
modelo = MultinomialNB()
modelo.fit(dados, marcacoes)
resultado = modelo.predict(misterioso)
print(resultado)
Alguém viu esse erro? sabe como solucionar?