多分类算法的python实现
时间: 2023-09-12 19:12:13 浏览: 97
各种排序算法的Python实现
多分类算法的Python实现有很多,以下列举几个常用的:
1. 决策树分类算法:使用sklearn库中的DecisionTreeClassifier,代码示例:
```python
from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
# 加载鸢尾花数据集
iris = load_iris()
# 取出数据特征和标签
X, y = iris.data, iris.target
# 训练模型
clf = DecisionTreeClassifier()
clf.fit(X, y)
# 预测
print(clf.predict(X))
```
2. K近邻分类算法:使用sklearn库中的KNeighborsClassifier,代码示例:
```python
from sklearn.neighbors import KNeighborsClassifier
from sklearn.datasets import load_iris
# 加载鸢尾花数据集
iris = load_iris()
# 取出数据特征和标签
X, y = iris.data, iris.target
# 训练模型
clf = KNeighborsClassifier()
clf.fit(X, y)
# 预测
print(clf.predict(X))
```
3. 朴素贝叶斯分类算法:使用sklearn库中的MultinomialNB或GaussianNB,代码示例:
```python
from sklearn.naive_bayes import MultinomialNB
from sklearn.datasets import fetch_20newsgroups
from sklearn.feature_extraction.text import CountVectorizer
# 加载新闻数据集
news = fetch_20newsgroups(subset='all')
# 特征提取
vectorizer = CountVectorizer()
X = vectorizer.fit_transform(news.data)
# 训练模型
clf = MultinomialNB()
clf.fit(X, news.target)
# 预测
print(clf.predict(X))
```
以上仅是部分多分类算法的Python实现,更多的算法和实现可以参考sklearn库中的文档。
阅读全文