在使用sklearn进行回归模型训练时,我们可以通过分配参数来获得每个模型的预测目标值。具体步骤如下:
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.tree import DecisionTreeRegressor
from sklearn.ensemble import RandomForestRegressor
X = 数据集的特征数据
y = 数据集的目标数据
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)
这里将数据集划分为80%的训练集和20%的测试集。
model1 = LinearRegression()
model2 = DecisionTreeRegressor()
model3 = RandomForestRegressor()
这里创建了三个回归模型对象,分别是线性回归模型、决策树回归模型和随机森林回归模型。
model1.fit(X_train, y_train)
y_pred1 = model1.predict(X_test)
model2.fit(X_train, y_train)
y_pred2 = model2.predict(X_test)
model3.fit(X_train, y_train)
y_pred3 = model3.predict(X_test)
分别使用训练集对每个模型进行训练,然后使用测试集进行预测,得到每个模型的预测目标值。
通过以上步骤,我们可以得到每个sklearn回归模型的预测目标值。根据具体的需求和数据特点,选择合适的回归模型进行训练和预测。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云