Numpy与Pandas实现多元线性回归:从基础到实践

勒令课程 2024-05-31 08:26:40
在数据科学与机器学习领域,线性回归作为一种基础而强大的预测模型,广泛应用于各种场景,从金融预测、市场营销到医疗健康分析,无不显现其身影。本文将引导您通过Python的两大数据分析神器——Numpy与Pandas,深入浅出地探索多元线性回归的理论与实践。 准备战:导入库与数据首先,确保您的环境中安装了Numpy与Pandas。在Python中,通过以下命令安装: pip install numpy pandas 然后,导入所需库,并准备数据。为简化演示,我们将使用Pandas内置的load_data函数加载示例数据集,或构造一个虚构的多元数据集: import numpy as npimport pandas as pdfrom sklearn.model_selection import train_test_splitfrom sklearn.linear_model import LinearRegressionfrom sklearn.metrics import mean_squared_error# 假设数据集已存在df中,包含特征X和目标变量ydf = pd.read_csv('your_data.csv') # 或构造数据集X = df[['feature1', 'feature2', 'feature3']] # 特征列y = df['target'] # 目标变量列# 划分训练集和测试集X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)构建模型:手动实现多元线性回归虽然Scikit-learn提供了现成的线性回归模型,但手动实现有助于深入理解其机制。线性回归模型公式为:y = β0 + β1x1 + β2x2 + … + βnxn + ε。我们使用Numpy实现: def manual_linear_regression(X, y): X_b = np.c_[np.ones((len(X), X)] theta_best = np.linalg.inv(X_b.T.dot(X_b)).dot(X_b.dot(y)) return theta_besttheta_best = manual_linear_regression(X_train, y_train)预测与评估有了模型参数后,我们就可以预测新数据,并使用均方误差(MSE)等指标评估模型: predictions = X_test @ theta_best[1:] + theta_best[0]mse = mean_squared_error(y_test, predictions)print(f'Mean Squared Error: {mse}')使用Pandas优化数据处理Pandas在数据预处理阶段尤为重要,它简化了数据清洗、特征工程等任务。比如,处理缺失值: df.fillna(df.mean(), inplace=True) # 用列均值填充缺失值进行特征缩放以提高模型性能: df_scaled = (df - df.mean()) / df.std() # 标准化深入微:与Scikit-learn集成尽管手动实现有教育意义,但在实际项目中,Scikit-learn的LinearRegression模型因其效率和便捷性更受欢迎: sklearn_model = LinearRegression().fit(X_train, y_train)predictions = sk_learn.predict(X_test)print(f'Scikit-learn MSE: {mean_squared_error(y_test, predictions)}')结语通过以上步骤,我们不仅亲手搭建了多元线性回归模型,还体验了Pandas在数据处理中的威力。实践是理解的最佳途径,希望本文激发您在实际项目中探索更复杂模型,深入数据科学的广袤天地。
0 阅读:3

勒令课程

简介:感谢大家的关注