相关疑难解决方法(0)

scikit 多标签分类:ValueError:输入形状错误

我beieveSGDClassifier()loss='log'支持多标记分类,我没有使用OneVsRestClassifier。检查这个

现在,我的数据集非常大,我正在使用HashingVectorizer并将结果作为输入传递给SGDClassifier. 我的目标有 42048 个特征。

当我运行它时,如下所示:

clf.partial_fit(X_train_batch, y)
Run Code Online (Sandbox Code Playgroud)

我得到:ValueError: bad input shape (300000, 42048)

我也使用类作为参数如下,但仍然是同样的问题。

clf.partial_fit(X_train_batch, y, classes=np.arange(42048))
Run Code Online (Sandbox Code Playgroud)

在 SGDClassifier 的文档中,它说 y : numpy array of shape [n_samples]

classification machine-learning stochastic-process scikit-learn

3
推荐指数
1
解决办法
3133
查看次数