Oi, Guilherme.
Segue o código, porém ainda não está rodando como deveria. Ele retorna 'sim' 'não'... deixei a dúvida em outro tópico...
import pandas as pd
df = pd.read_csv('busca.csv')
x_df = df[['home', 'busca', 'logado']]
y_df = df['comprou']
x_dummies = pd.get_dummies(x_df['busca'])
y_dummies = y_df
x = x_dummies.values
y = y_dummies.values
porcentagem_de_treino = 0.9
tamanho_de_treino = int(porcentagem_de_treino * len(y))
tamanho_de_teste = len(y) - tamanho_de_treino
treino_dados = x[:tamanho_de_treino]
treino_marcacoes = y[:tamanho_de_treino]
teste_dados = x[-tamanho_de_teste:]
teste_marcacoes = y[-tamanho_de_teste:]
from sklearn.naive_bayes import MultinomialNB
modelo = MultinomialNB
modelo.fit(treino_dados, treino_marcacoes)
resultado = modelo.predict(teste_dados)
diferencas = resultado - teste_marcacoes
acertos = [d for d in diferencas if d==0]
total_de_acertos = len(acertos)
total_de_elementos = len(teste_dados)
taxa_de_acerto = 100.0 * total_de_acertos/ total_de_elementos
print(taxa_de_acerto)
print(total_de_elementos)