【发布时间】:2021-04-16 17:10:00
【问题描述】:
import numpy as np
import pandas as pd
import seaborn as sns
from sklearn.linear_model import LinearRegression
from sklearn.datasets import make_regression
a = make_regression(n_samples=300,n_features=5,noise=5)
df1 = pd.DataFrame(a[0])
df1 = pd.concat([df1,pd.DataFrame(a[1].T)],axis=1,ignore_index=True)
df1.rename(columns={0:"X1",1:"X2",2:"X3",3:"X4",4:"X5",5:"Target"},inplace=True)
sns.heatmap(df1.corr(),annot=True);
现在我可以问我的问题了。如何选择将包含在模型中的特征?
【问题讨论】:
-
您能向我们展示您使用的部分数据集吗?\
-
Dataset Head 它只是一个来自 make_regression 的随机数据集。
标签: linear-regression feature-selection