python训练模型并进行分类
时间: 2023-10-25 20:09:50 浏览: 48
Python有很多机器学习框架可以用来训练模型和进行分类,比如Scikit-learn、TensorFlow、Keras等等。
下面是一个简单的例子,使用Scikit-learn框架来训练一个逻辑回归模型并进行分类:
```python
# 导入依赖库
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.datasets import load_iris
# 加载鸢尾花数据集
iris = load_iris()
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=0.2, random_state=42)
# 训练逻辑回归模型
clf = LogisticRegression(random_state=0, multi_class='ovr').fit(X_train, y_train)
# 在测试集上进行预测
y_pred = clf.predict(X_test)
# 输出分类准确率
print("Classification accuracy: ", clf.score(X_test, y_test))
```
这个例子中,我们使用了Scikit-learn的LogisticRegression类来训练一个逻辑回归模型,加载了经典的鸢尾花数据集,将数据集划分为训练集和测试集,最后在测试集上进行预测并输出分类准确率。