Ao tentar executar o código está ocorrendo erro "operands could not be broadcast together with shapes (100,) (100,5)".
df = pd.read_csv("busca.csv")
df.head()
a_trocar = {
'sim': 1,
'nao': 0
}
X_df = df[["home", "busca", "logado"]]
Y_df = df["comprou"]
Xdummies_df = pd.get_dummies(X_df)
Ydummies_df = Y_df.astype(int)
X = Xdummies_df.values
Y = Ydummies_df.values
treino_X = X[:900]
treino_Y = Y[:900]
teste_X = X[-100:]
teste_Y = X[-100:]
print("Treinaremos com %d elementos e testaremos com %d elementos" % (len(treino_X), len(teste_X)))
from sklearn.naive_bayes import MultinomialNB
modelo = MultinomialNB()
modelo.fit(treino_X, treino_Y)
resultado = modelo.predict(teste_X)
diferencas = resultado - teste_Y`
