Python 机器学习库是一个强大的工具,可以帮助开发者构建复杂的机器学习模型。以下是一些常用的Python机器学习库:

常用库

  • Scikit-learn:这是一个非常流行的机器学习库,提供了各种机器学习算法的实现。
  • TensorFlow:Google 开发的一个开源机器学习框架,适用于深度学习。
  • PyTorch:Facebook 开发的一个开源机器学习库,同样适用于深度学习。

示例代码

以下是一个使用Scikit-learn进行简单线性回归的示例:

from sklearn.linear_model import LinearRegression
import numpy as np


X = np.array([[1], [2], [3], [4], [5]])
y = np.dot(X, np.array([1.0, 2.0])) + 3

# 创建线性回归模型
model = LinearRegression()

# 训练模型
model.fit(X, y)

# 预测
print(model.predict(np.array([[6]])))

扩展阅读

想要了解更多关于Python机器学习的知识,可以访问我们的Python机器学习教程

图片

Scikit-learn 图标
TensorFlow 图标
PyTorch 图标