scikit-learn - sklearn.linear_model.SGDClassifier 手动推理多类分类

Question

我已经使用 sklearn.linear_model.SGDClassifier 训练了一个用于 3 类分类的模型。现在我正在寻找一种手动推断模型的方法。这里的问题是模型包含三对 [coef_, intercept_] 所以我不明白如何在 C++ 中进行预测。

训练代码类似于 sklearn 示例：

clf = make_pipeline(StandardScaler(), SGDClassifier(max_iter=1000, tol=1e-3))
clf.fit(train_features, train_labels)

我试图计算coef_ * sample + intercept_每个类的值，但不明白如何使用这些数字确定类。

标签： scikit-learnclassificationlinear-regressionmulticlass-classification