scikit-learn

scikit-learn

Machine Learning in Python

预测数据分析的 Python 基石

scikit-learn 提供简单高效的工具,建立在 NumPy、SciPy、matplotlib 之上,开源且可商用(BSD 协议),是数据科学与传统 ML 的首选入口。

2007
项目起始(Google Summer of Code)
BSD-3
开源协议(可商用)
NumPy·SciPy
底层依赖基石

免费试用 · 无需下载即可在线体验

人人可用的机器学习

scikit-learn

scikit-learn 由 David Cournapeau 于 2007 年作为 Google Summer of Code 项目发起,此后由社区志愿者持续维护。它提供一套简单、高效、可复用的工具,用于预测数据分析,覆盖分类、回归、聚类、降维、模型选择与预处理等典型任务。

其设计哲学是「一致且易上手的 API」:fit / predict / transform 的统一接口让你在不同算法间无缝切换;同时它构建在 NumPy、SciPy 与 matplotlib 之上,与整个科学计算生态天然兼容,既适合入门学习,也能支撑生产级管线。

核心功能一览

六大核心能力,覆盖创作全流程

🔍

分类

垃圾邮件检测、图像识别等,提供梯度提升、KNN、随机森林、逻辑回归等算法。

📈

回归

预测连续值属性,如药物反应、股价,支持梯度提升、随机森林、岭回归等。

🧮

聚类

自动把相似对象分组,用于客户分群、实验结果归类等无监督场景。

📉

降维

PCA、特征选择、非负矩阵分解等方法,用于可视化与提升训练效率。

🛠️

模型选择与预处理

网格搜索、交叉验证、指标评估,以及文本/数值特征提取与归一化。

零基础快速上手

4 步完成你的第一件 AI 作品,无需任何设计经验

1

安装 scikit-learn

确保已有 NumPy/SciPy,再用 pip install -U scikit-learn 或 conda install -c conda-forge scikit-learn。

2

准备数据

用 NumPy 数组或 Pandas 组织特征与标签,并按需做 train/test 切分。

3

训练与预测

实例化 estimator 后调用 fit 训练、predict 预测,统一接口贯穿所有算法。

4

评估与调参

用 cross_val_score、GridSearchCV 做验证与超参搜索,再用 metrics 评估效果。

全平台客户端下载

支持网页端、桌面端、移动端及浏览器插件,随时随地开启创作

官方与文档

scikit-learn 官网
首页与教程
下载
用户指南
系统性文档
下载
GitHub 仓库
源码与贡献
下载

生态

PyPI 安装包
pip 发布页
下载
示例集
可运行范例
下载
SpeedAIAD

常见问题解答

遇到问题?这里可能有你想要的答案

1

scikit-learn 能训练深度学习模型吗?

scikit-learn 聚焦于传统机器学习与浅层模型,不擅长大规模神经网络;深度学习任务通常交给 PyTorch、TensorFlow 或 Keras,scikit-learn 可负责前处理与评估环节。

2

scikit-learn 和商业使用冲突吗?

不冲突。scikit-learn 采用 3-Clause BSD 协议,允许商业使用、修改与再分发,只需保留版权声明即可。

与主流机器学习库对比

和同类工具横向对比,帮你选对工具。

对比维度scikit-learnXGBoostPyTorch
定位通用 ML 工具集梯度提升专用深度学习框架
代表任务分类/回归/聚类/降维表格数据梯度提升神经网络训练
上手成本低,API 一致中,需理解 boosting高,需理解训练循环
底层依赖NumPy / SciPy / matplotlib自包含 C++ 核心torch / CUDA