Как сделать таблицу или DataFrame из dtype = object? - PullRequest
Новая
       23

Как сделать таблицу или DataFrame из dtype = object?

0 голосов
/ 02 января 2019

Я хотел бы показать, как делаются прогнозы в таблице или в DataFrame.

Я попытался поместить X_test, y_test и предсказания (gnations = model.predict (X_test)) в DataFrame, чтобы показать, чтоотзывы положительные или отрицательные. 

import pandas as pd

predictions = model.predict(X_test)
df_prediction = pd.DataFrame({
    'X_test': [X_test],
    'y_test': [y_test],
    'prediction': [predictions]
})
df_prediction.head()

Переменная «X_test» (имя: обзор, длина: 4095, dtype: object) выглядит следующим образом:

15806 ['tyre', 'gauges',' kind ',' thing ',' make ',' ...

541 ['like', 'said', 'title', 'review', 'say', 'pr ...

...

Переменная "y_test" (имя: метка, длина: 4095, dtype: объект) выглядит следующим образом:

15806 позитив

541 positiv

...

Переменная «предсказания» выглядит следующим образом:

array (['positiv', 'positiv', 'positiv', ..., 'positiv ',' positiv ',' positiv '], dtype = object)

На данный момент я получил DataFrame со всеми данными в первой строке, но мне нужно, как таблица со всеми строками.

1 Ответ

0 голосов
/ 02 января 2019

Если x_test, y_test и predictions являются списками, то вы можете просто сделать это: 

df_prediction = pd.DataFrame({
    'X_test': x_test,
    'y_test': y_test,
    'prediction': predictions
})

Также, df_prediction.head() напечатает первые 5 строк кадра данных.Вы можете использовать print(df_prediction) или просто df_prediction (для форматированного вывода в Jupyter Notebook), чтобы увидеть все данные.

...