sklearn实现一个分类算法
时间: 2024-01-04 08:03:49 浏览: 75
sklearn中的KNN算法实现鸢尾花分类.py
好的,您可以使用sklearn中的SVM(支持向量机)算法进行分类。以下是一个简单的SVM分类器实现示例:
```
from sklearn import svm
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
# 加载数据集
iris = load_iris()
X = iris.data
y = iris.target
# 分割数据集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=0)
# 创建SVM分类器
clf = svm.SVC(kernel='linear', C=1.0)
# 训练模型
clf.fit(X_train, y_train)
# 预测测试集
y_pred = clf.predict(X_test)
# 输出准确率
print("Accuracy:", metrics.accuracy_score(y_test, y_pred))
```
在这个示例中,我们使用鸢尾花数据集作为示例数据,并将其分成训练集和测试集。然后我们使用SVM分类器进行训练,并预测测试集。最后,我们输出预测准确率。您可以根据自己的数据集和具体需求进行调整和修改。
阅读全文