Python PM2.5 预测模型:结合温度和湿度
建立 PM2.5 预测模型:
- 导入相关库
import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error
- 读取数据,并进行预处理
data = pd.read_csv('data.csv')
data = data.dropna() # 删除含有空值的行
X = data[['temperature', 'humidity']]
y = data['PM2.5']
- 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
- 建立线性回归模型并训练
model = LinearRegression()
model.fit(X_train, y_train)
- 预测 PM2.5 值并评估模型
y_pred = model.predict(X_test)
mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
print('RMSE:', rmse)
建立温度和湿度预测模型与上述步骤相似,只需改变 y 值为温度或湿度即可。
原文地址: https://www.cveoy.top/t/topic/lL3x 著作权归作者所有。请勿转载和采集!