Solucionado (ver solução)
Solucionado
(ver solução)
3
respostas

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

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

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

solução!
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()
)

Obrigada, funcionou!

Quer mergulhar em tecnologia e aprendizagem?

Receba a newsletter que o nosso CEO escreve pessoalmente, com insights do mercado de trabalho, ciência e desenvolvimento de software