支持向量机SVM推理(预测)代码实现

使用sklearn训练的SVM模型要怎样移植到C语言中进行嵌入式开发呢?本文提供了将训练的支持向量机模型从sklearn抽取出来的方法。

clf = SVC(kernel='rbf', C=10.0, gamma=0.01)
    
clf.fit(X_train, y_train)

# 使用sklearn自带的predict函数
y_predict = clf.predict(X_validate)
print(confusion_matrix(y_validate, y_predict))


# 使用训练好的SVC模型中的clf.support_vectors_, clf.intercept_,clf.dual_coef_和clf.gamma参数,
# 根据核函数类型实现预测功能
fp = np.zeros((X_validate.shape[0], 1))
for i in range(X_validate.shape[0]):
    K = np.exp(-np.sum((X_validate[i, :] - clf.support_vectors_) * (X_validate[i, :] - 
    clf.support_vectors_), 1) * clf.gamma)
    fp[i] = np.dot(clf.dual_coef_, K) + clf.intercept_
fps = fp.squeeze()
fps[fps >= 0] = 1
fps[fps < 0] = 0
print(confusion_matrix(y_validate, fps))

这里使用的是径向基函数rbf核函数,X_train和y_train是训练样本的输入特征和标签,X_validate和y_validate是测试集的输入特征和标签,clf是训练的SVC模型。

你可能感兴趣的:(机器学习,支持向量机,机器学习,推理实现,SVM移植,SVM预测)