我想要达到的结果
scikit-learn有什么用?怎么用呢?什么场景下用
以下内容由CHATGPT及阿里嘎多学长共同生成、有用望采纳:
scikit-learn是一个基于Python的开源机器学习库,它在机器学习和人工智能领域有着广泛的应用。以下是关于scikit-learn的主要用途、安装和使用方式,以及适用场景的详细介绍:
安装scikit-learn非常简单,可以通过Python的包管理工具pip进行安装:
pip install scikit-learn
如果下载速度过慢,可以使用国内的镜像源,例如清华镜像源:
pip install scikit-learn -i https://pypi.tuna.tsinghua.edu.cn/simple
安装完成后,可以在Python程序中导入并使用:
import sklearn
# 其他代码...
# 导入所需的库和模块
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
# 读取数据集
data = pd.read_csv("housing.csv")
# 划分数据集为训练集和测试集
X = data.drop("price", axis=1)
y = data["price"]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建并训练模型
model = LinearRegression()
model.fit(X_train, y_train)
# 预测
y_pred = model.predict(X_test)
# 评估模型
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)
print(f"Mean Squared Error: {mse}")
print(f"R-Squared: {r2}")