首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

浅谈Entity Embedding

本文将简单介绍利用神经网络来表示类别特征的方法-Entity Embedding,这个方法首先出现在kaggle上的《Rossmann Store Sales》中的rank 3的解决方案,作者在比赛完后为此方法整理一篇论文放在了...arXiv,文章名:《Entity Embeddings of Categorical Variables》。...target encoding 特征无内在顺序,category数量 > 4, K-fold cross validation 不做处理(模型自动编码) CatBoost,lightgbm 2 实体嵌入 Entity...在《Entity Embeddings of Categorical Variables》 结构非常简单,就是embedding层后面接上了两个全连接层,代码用keras写的,构建模型的代码量也非常少,...代码实践 作者代码 : https://github.com/entron/entity-embedding-rossmann 自己的尝试:https://github.com/yanqiangmiffy

2.1K20
领券