框架

机器学习、深度学习、强化学习领域的经典算法与主流框架

机器学习

传统 ML

Scikit-learn

最流行的 Python 机器学习库,分类、回归、聚类等经典算法。

访问官网 →
梯度提升

XGBoost

高效梯度提升树,Kaggle 竞赛利器,支持分类与回归。

访问官网 →
梯度提升

LightGBM

微软开源梯度提升框架,训练快、内存低、支持大规模数据。

访问官网 →
梯度提升

CatBoost

Yandex 梯度提升库,原生支持类别特征,无需预处理。

访问官网 →

深度学习

框架

PyTorch

Meta 开源深度学习框架,动态计算图,研究与生产首选。

访问官网 →
框架

TensorFlow

Google 全栈深度学习平台,支持 Keras API 与生产部署。

访问官网 →
框架

JAX

Google 高性能数值计算库,自动微分 + JIT 编译,适合科研。

访问官网 →
框架

Keras

高层神经网络 API,简洁易用,支持 TF / JAX / PyTorch 后端。

访问官网 →

强化学习

环境

OpenAI Gym

标准 RL 环境库,丰富的仿真环境用于算法开发与对比。

访问官网 →
框架

Stable Baselines3

基于 PyTorch 的可靠 RL 算法实现,开箱即用。

访问官网 →
框架

Ray RLlib

可扩展分布式 RL 框架,支持多算法大规模并行训练。

访问官网 →

计算机视觉

OpenCV

最流行的计算机视觉库,图像处理、特征检测、视频分析。

访问官网 →
检测

YOLO

实时目标检测算法系列,速度快、精度高,适合工程落地。

访问官网 →
分割

Detectron2

Meta 开源检测与分割框架,基于 PyTorch。

访问 GitHub →

自然语言处理

Hugging Face Transformers

最流行预训练模型库,BERT / GPT / T5 等数千种模型。

访问官网 →

spaCy

工业级 NLP 库,高效处理分词、NER、依存句法分析。

访问官网 →

NLTK

经典 NLP 工具包,适合教学与研究,丰富语料库。

访问官网 →