import numpy as np import pandas as pd # 标签编码 from sklearn.preprocessing import LabelEncoder # 随机森林回归模型 from sklearn.ensemble import RandomForestRegressor # 交叉验证 from sklearn.model_selection import cross_val_score data = pd.read_excel('./data/汽车数据集/car.xlsx') le = LabelEncoder() for i in data.columns: data[i] = le.fit_transform(data[i]) from sklearn.model_selection import train_test_split train_x, test_x, train_y, test_y = train_test_split(data.iloc[:, :-1], data.iloc[:, -1], random_state=7) model = RandomForestRegressor(max_depth=6, n_estimators=200, random_state=7) model.fit(train_x, train_y) cvs = cross_val_score(model, train_x, train_y, cv=5, scoring='f1_weighted') print('f1得分: ', cvs.mean())
时间: 2023-12-13 10:03:58 浏览: 109
Random-Forest-Regression:森林随机回归
5星 · 资源好评率100%
这段代码使用了随机森林回归模型对汽车数据集进行了训练和交叉验证。首先,使用pandas读取了Excel格式的汽车数据集,并使用LabelEncoder对所有特征进行了标签编码。然后,使用train_test_split将数据集分成了训练集和测试集。接着,使用随机森林回归模型对训练集进行了训练,并使用交叉验证对模型进行了评估。最后,使用f1_weighted作为评估指标,计算了模型的平均f1得分,并输出了结果。
阅读全文