Pandas: объединение фактических и прогнозных значений

Question

Я пытаюсь объединить фактические целевые значения и прогнозируемое целевое значение в виде фрейма данных. Однако я получаю следующую ошибку. Не уверен, почему это происходит.

a = pd.DataFrame(y_test, columns=['Actual'])
b = pd.DataFrame(final_model.predict(X_test), columns=['Predictions'])
c = pd.concat([a, b])
c.head()

Фактическое значение по сравнению с прогнозом

Answer 1

@ nimbous уже ответил, но если вам нужно меньше промежуточных шагов, используйте словарь для создания df.

c = pd.DataFrame({"Actual": y_test, "Predictions": final_model.predict(X_test)})

Answer 2

Попробуйте с pd.concat([a, b], axis=1), хотя это не лучшее решение.

Answer 3

Вы можете напрямую назначить новый столбец. Пока форма y_test и прогнозы модели совпадают, у вас не должно возникнуть проблем с выполнением следующего:

base = pd.DataFrame()
base["actual"] = y_test
base["predictions"] = final_model.predict(X_test)