我一直在尝试使用 scikit-learn 进行回归,但存在这样的多个输出问题:
X = np.random.random((10,3))
y = np.random.random((10,2))
X2 = np.random.random((7,3))
clf = SVR()
clf.fit(X, y)
y_pred = clf.predict(X2)
问题是这行不通。它失败了:
ValueError: Buffer has wrong number of dimensions (expected 1, got 2)
有谁知道如何在 scikit-learn 中处理具有多个输出的回归?
编辑。我注意到 RandomForestRegressor、KNeighborsRegressor 和 LinearRegression 都可以工作,但 rf 是唯一一个在我的数据集上接近于好的方法!有什么适合的方法SVR吗gbm?