преобразовать числовую функцию в имена объектов обратно в SelectKBest - PullRequest
Новая
       19

преобразовать числовую функцию в имена объектов обратно в SelectKBest

0 голосов
/ 29 марта 2020 
data = pd.read_csv('Book1.csv', usecols=['tokenize'])
#ganti jadi 'pre_pro/data1.csv'

#TFIDF VECTORIZER
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.feature_selection import SelectKBest, chi2

data_sample = data[2:4]
tfidf= TfidfVectorizer(smooth_idf=True,norm=None)
X = tfidf.fit_transform(data_sample['tokenize'])

data_sample = pd.DataFrame(X.toarray(), columns=tfidf.get_feature_names())
y = data['tokenize'][2:4]
X_new = SelectKBest(chi2, k=5).fit_transform(X,y)
X_new = pd.DataFrame(X_new.toarray(),y)

в этом коде X_new дает мне только заголовки с числовой функцией, поэтому я не вижу, какая функция появляется. я хочу, чтобы числа (0,1,2,3,4) в этой строке стали именами элементов. 

|            tokenize           | 0    | 1    | 2    | 3    | 4    |
|:-----------------------------:|------|------|------|------|------|
| it was so bad the car cant on | 1.40 | 1.40 | 1.40 |  0   | 1.40 |
| car is very good indeed       | 0    |    0 |    0 | 1.40 | 0    |
...