python svm 输出概率
时间: 2023-06-29 22:07:29 浏览: 86
svm_gui_svm预测_pythonGUI编程_SVM_
在Python的sklearn库中,SVM分类器的输出概率可以通过设置参数`probability=True`来实现。具体的实现步骤如下:
1. 导入SVM分类器模型及相关的库:
```python
from sklearn.svm import SVC
```
2. 定义SVM分类器并设置参数`probability=True`:
```python
clf = SVC(kernel='linear', probability=True)
```
3. 使用训练数据进行训练:
```python
clf.fit(X_train, y_train)
```
4. 预测测试数据,并输出概率:
```python
y_pred = clf.predict(X_test)
prob = clf.predict_proba(X_test)
```
其中,`y_pred`表示预测结果,`prob`表示输出概率。注意,`predict_proba()`方法返回的是一个二维数组,第一列代表预测为负样本的概率,第二列代表预测为正样本的概率。
希望能对你有所帮助!
阅读全文