使用 scikit-learn 中的 SVM 分类器对乳腺癌威斯康星州数据集进行分类。
时间: 2023-11-26 19:11:42 浏览: 171
好的,我可以帮你解答这个问题。
首先,需要准备数据集。乳腺癌威斯康星州数据集已经内置在 scikit-learn 中,可以直接调用。代码如下:
```python
from sklearn.datasets import load_breast_cancer
data = load_breast_cancer()
X = data.data
y = data.target
```
接着,可以将数据集分为训练集和测试集,以便评估模型的性能。我们可以使用 train_test_split 函数来完成这个任务。代码如下:
```python
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
```
然后,我们可以使用 SVM 分类器来训练模型。这里使用 scikit-learn 中的 SVC 类来实现。代码如下:
```python
from sklearn.svm import SVC
clf = SVC(kernel='linear')
clf.fit(X_train, y_train)
```
在训练完成后,我们可以使用测试集来评估模型的性能。这里使用 accuracy_score 函数来计算分类器的准确率。代码如下:
```python
from sklearn.metrics import accuracy_score
y_pred = clf.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)
```
上述代码输出的结果即为 SVM 分类器在测试集上的准确率。
希望这个回答能够帮助到你。
阅读全文