Código do tfidf não rodou no colab | Linguagem Natural parte 2: continuando com a análise de sentimento

tfidf_tratado = tfidf.fit_transform(resenha['tratamento_5']) treino, teste, classe_treino, classe_teste = train_test_split(tfidf_tratado, resenha['tratamento_5'], random_state = 42) regressao_logistica = LogisticRegression() regressao_logistica.fit(treino, classe_treino) acuracia_tfidf_tratado = regressao_logistica.score(teste, classe_teste) print(acuracia_tfidf_tratado)

Oii Edgard, tudo bem contigo?

Acredito que seu código não esteja rodando devido a um erro. Na segunda linha, na qual você está dividindo a base de dados em treino e teste, onde você colocou resenha['tratamento_5'] deveria ser resenha['classificacao']. Dessa forma, o código fica da seguinte maneira:

tfidf_tratado = tfidf.fit_transform(resenha['tratamento_5'])
treino, teste, classe_treino, classe_teste = train_test_split(tfidf_tratado,
                                                              resenha['classificacao'],
                                                              random_state = 42)

regressao_logistica = LogisticRegression() 
regressao_logistica.fit(treino, classe_treino)
acuracia_tfidf_tratado = regressao_logistica.score(teste, classe_teste)

print(acuracia_tfidf_tratado)

Tenta fazer essa correção e depois me conta aqui se funcionou certinho ^^

Espero que isso ajude, qualquer dúvida estou à disposição :)

Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!