Python代码实现测试训练良性恶性肿瘤数据,直接从sklearn导入如下 from sklearn.datasets import load_breast_cancer cancer = load_breast_cancer()
时间: 2023-09-03 12:14:58 浏览: 132
基于Python实现乳腺癌症状数据挖掘(源码+数据+说明文档).rar
5星 · 资源好评率100%
好的,以下是一个简单的 Python 代码示例,用于测试和训练良性恶性肿瘤数据:
```python
# 导入必要的库
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score
# 加载良性恶性乳腺肿瘤数据集
cancer = load_breast_cancer()
# 将数据集拆分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(cancer.data, cancer.target, test_size=0.3, random_state=42)
# 创建支持向量机分类器并拟合训练数据
svm = SVC(kernel='linear', C=1, random_state=42)
svm.fit(X_train, y_train)
# 预测测试数据集的分类标签
y_pred = svm.predict(X_test)
# 计算模型的准确性
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)
```
这段代码将数据集加载到 `cancer` 变量中,然后使用 `train_test_split()` 函数将数据集拆分为训练集和测试集。接着创建一个支持向量机分类器,并使用 `fit()` 函数拟合训练数据。最后,使用 `predict()` 函数预测测试数据集的分类标签,并使用 `accuracy_score()` 函数计算模型的准确性。
阅读全文