Entrar Ainda não tem acesso?
Solucionado (ver solução)
Solucionado
(ver solução)
3
respostas

[Dúvida] AttributeError: 'TfidfVectorizer' object has no attribute 'get_feature_names'

Referente ao curso Linguagem Natural parte 2: continuando com a análise de sentimento, no capítulo TF-IDF e Ngrams: técnicas mais avançadas e atividade Aplicando o TF-IDF

por Estudante
| 59.1k xp | 3 posts

Olá,

Ao aplicar o TF-IDF, no exemplo dado em vídeo:

from sklearn.feature_extraction.text import TfidfVectorizer

frases = ["Assisti um filme ótimo", "Assisti um filme péssimo"]

tfidf = TfidfVectorizer(lowercase=False, max_features=50)

caracteristicas = tfidf.fit_transform(frases)
pd.DataFrame(
    caracteristicas.todense(),
    columns=tfidf.get_feature_names()
)

É retornado o seguinte erro:

AttributeError: 'TfidfVectorizer' object has no attribute 'get_feature_names'

Como resolver? Estou usando o Google Colab.

Abs!

3 respostas
por Guilherme Alves Silveira
| 35.5k xp | 1 posts
Analista de Sistemas Senior

Olá. Tem que usar o método tfidf.get_feature_names_out()

solução!
por Marcelo De Araújo
| 171.9k xp | 16 posts
Desenvolvedor iOS 
from sklearn.feature_extraction.text import TfidfVectorizer
import pandas as pd

frases = ["Assisti um filme ótimo", "Assisti um filme péssimo"]

tfidf = TfidfVectorizer(lowercase=False, max_features=50)

caracteristicas = tfidf.fit_transform(frases)
pd.DataFrame(
    caracteristicas.todense(),
    columns=tfidf.vocabulary_.keys()
)
por Estudante
| 59.1k xp | 3 posts

Obrigada, funcionou!