常用机器学习库和深度学习库整理

本文同步发表于个人博客 https://wz.anoms.top/2025/01/19/Common-used-machine-learning-library/

这一年多以来的科研工作中，调研了不少机器学习库的相关知识，临近年底，事情不多，感觉正好可以整理一下。由于笔者主要使用python的缘故，这里介绍的函数库以python库居多，但也顺带介绍了一些其他编程语言的机器学习库。

python库

scikit-learn

官网： https://scikit-learn.org/

学过统计学和机器学习的同学们对这个库应该不陌生。这个库提供了许多统计学习算法的API，几乎涵盖了机器学习的绝大多数方面（如下图），包括线性回归、逻辑回归、PCA、KNN、决策树等，甚至还支持搭建比较简单的神经网络。

Pytorch

官网： https://pytorch.org/

这个库的前身是torch，一个使用Lua语言实现的深度学习库，其在2010年前后比较热门。后来，Facebook母公司Meta基于这个库，使用C++等编程语言重写了核心代码，并基于 Python 生态开发了 PyTorch，使其更适合研究和工业应用。目前PyTorch是最流行的深度学习库之一。几年前其排名已经超过了TensorFlow。Pytorch相比于TensorFlow的一个优点在于其对动态图计算（Dynamic Computati