Datawhale AI 夏令营——电力需求预测赛
最后在应用随机森林回归模型的时候,函数rf.fit里面的内容必须是整数,但文件中“id”是字符集,因此没有选择此特征列。首先需要提取训练集和需要预测的特征列,这里是“dt”,“type”,然后根据时间的判断条件进一步提取“target”特征列的数值,然后得到测试集对应的特征列,之后如果想要验证结果的准确性,可以从训练集里面通过train_test_split函数提取验证集以及剩下的训练集,最后对剩余的训练集应用随机森林回归模型,最终得到“target”的预测结果并生成对应格式文件。# 划分训练集和验证集。
原创
2024-07-14 13:05:35 ·
534 阅读 ·
1 评论