未来杯城市-房产租金预测开源baseline

青木长风

于 2019-05-06 12:06:55 发布

阅读量1.4k

点赞数 1

分类专栏：数据挖掘文章标签：数据挖掘 baseline 未来杯房租预测

本文链接：https://blog.youkuaiyun.com/qq_39936389/article/details/89879738

版权

团队公开了在'未来杯'比赛中实现0.85线上评分的baseline方案，详细代码和说明可在知乎专栏及GitHub找到。主要采用了lgb模型进行预测。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

昨天跟队友开源了一份线上0.85的baseline，发在我们的知乎专栏里了。
知乎地址：
https://zhuanlan.zhihu.com/p/64715267

github地址：
https://github.com/leo6033/future-AI-challenge

下面贴一下baseline的主要代码部分：

def parseData(df):
    """
    预处理数据
    """
    df['rentType'][df['rentType']=='--'] = '未知方式'
    # 转换object类型数据
    columns = ['rentType', 'houseFloor', 'houseToward', 'houseDecoration', 'communityName', 'region', 'plate']
    for col in columns:
        df[col] = df[col].astype('category')
        
    # 将buildYear列转换为整型数据
    tmp = df['buildYear'].copy()
    tmp2 = tmp[tmp!='暂无信息'].astype('int')
    tmp[tmp=='暂无信息'] = tmp2.mo