【发布时间】:2018-11-30 11:09:34
【问题描述】:
我想建立一个模型来预测将租用多少辆自行车。因此我正在使用 DecisionTreeRegressor 但我遇到了问题。我无法绘制这个,因为“ValueError:x 和 y 必须是相同的大小”。如果我将 X 和 y 与 len() 进行比较,我得到相同的长度。问题是我的 X 有 10 列而 y 只有 1 列。这就是我收到此错误的原因。 我想用我的情节得到以下图片:
有什么办法可以纠正这个问题吗?
X = df_day.drop(["date", "total_rides"], axis = 1)
y = df_day["total_rides"]
#X has something like 10 columns and y has only 1 column
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state = 43, test_size = 0.25)
dtc = DecisionTreeRegressor(max_depth = None)
dtc.fit(X_train, y_train)
#dtc.score(X_test, y_test)
#Prediction
y_pred2 = dtc.predict(X_test)
plt.scatter(X, y)
plt.plot(X_test, y_pred2)
plt.show()
【问题讨论】:
-
如果 x 有 10 列,你如何在 2d 图中将它与 y 绘制?你想通过绘制这些数据来实现什么?
-
我仍然无法理解您的问题到底是什么。你能否修改示例数据来说明你得到了什么以及你想要什么? (为简单起见,可能不是 10 天,而是 3 天,不是 24 小时,而是 4 小时)
标签: python matplotlib data-science