阅读量:0
在Python中，有多个机器学习库可供选择，包括Scikit-learn、TensorFlow、Keras、PyTorch等。以下是一个简单的示例，演示如何在Python中使用Scikit-learn库来训练和预测一个简单的线性回归模型：
# 导入所需的库 from sklearn import linear_model import numpy as np # 创建一些示例数据 X = np.array([[1], [2], [3], [4], [5]]) y = np.array([2, 4, 6, 8, 10]) # 创建线性回归模型 model = linear_model.LinearRegression() # 训练模型 model.fit(X, y) # 进行预测 prediction = model.predict([[6]]) print(prediction) 这只是一个简单的例子，演示了如何使用Scikit-learn库中的线性回归模型。您可以根据您的需求选择合适的机器学习库，并使用它们来实现更复杂的机器学习任务。
