Scikit-learn 是一个开源的机器学习库,旨在提供简单高效的工具,以方便数据科学家进行数据挖掘和数据分析。以下是一些关于 Scikit-learn 的基本文档和资源。

安装

您可以通过以下命令安装 Scikit-learn:

pip install scikit-learn

快速开始

  1. 导入库
from sklearn import datasets
  1. 加载数据集
iris = datasets.load_iris()
  1. 创建模型
from sklearn.ensemble import RandomForestClassifier

clf = RandomForestClassifier()
  1. 训练模型
clf.fit(iris.data, iris.target)
  1. 预测
prediction = clf.predict([[5.1, 3.5, 1.4, 0.2]])

资源

示例图片

Scikit-learn Logo

返回首页