本文同步发表于个人博客 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++等编程语言重写了核心代码