如何加入DecisionTreeRegressor预测输出到原始数据

时间:2016-04-26 00:05:25

标签: join merge concat decision-tree predict

我正在开发一个使用DecisionTreeRegressor的模型。我使用训练数据构建并拟合了树,并预测了最新数据的结果,以确认模型的准确性。

构建并适合树:     X = np.matrix(pre_x)     y = np.matrix(pre_y)     regr_b = DecisionTreeRegressor(max_depth = 4)     regr_b.fit(X,y)

预测新数据:     X = np.matrix(pre_test_x)     trial_pred = regr_b.predict(X,check_input = True)

trial_pred是预测值的数组。我需要将它连接回pre_test_x,以便我可以看到预测与实际发生的情况相符。

我尝试过合并:

all_pred = pre_pre_test_x.merge(predictions, left_index = True, right_index = True)

all_pred = pd.merge (pre_pre_test_x, predictions, how='left', left_index=True, right_index=True  )

并且在所有现有列中都没有得到任何结果或附加到DataFrame底部的列的第二个副本。

1 个答案:

答案 0 :(得分:1)

事实证明这很简单。将预测输出保留为数组,然后运行:     w_pred = pre_pre_test_x.copy(deep = True)     w_pred [' pred_val'] = trial_pred