Scikit-learn 是一个开源的机器学习库,旨在提供简单高效的工具,以方便数据科学家进行数据挖掘和数据分析。以下是一些关于 Scikit-learn 的基本文档和资源。
安装
您可以通过以下命令安装 Scikit-learn:
pip install scikit-learn
快速开始
- 导入库:
from sklearn import datasets
- 加载数据集:
iris = datasets.load_iris()
- 创建模型:
from sklearn.ensemble import RandomForestClassifier
clf = RandomForestClassifier()
- 训练模型:
clf.fit(iris.data, iris.target)
- 预测:
prediction = clf.predict([[5.1, 3.5, 1.4, 0.2]])
资源
Scikit-learn 官方文档 - 英文文档,提供了详细的安装指南、教程和API文档。
Scikit-learn 中文文档 - 中文文档,适合中文用户阅读。
示例图片
Scikit-learn Logo