首页 > 解决方案 > 任何改进神经网络对特征月的泛化的方法

问题描述

我想开发一个神经网络来预测冠状病毒的感染率。我有一个功能monthOfYear,并使用一个输入节点来表示它。该节点可以采用从 1 到 12 的整数值,表示一年中的月份。然而,结果表明该神经网络的泛化能力较差,尤其是在一年的十二月和次年的一月之间。对设计monthOfYear的新表示有什么建议吗?谢谢。

标签: machine-learningneural-networkconv-neural-network

解决方案


当表示一个类(比如一个月)时,你应该避免使用整数,而是使用OneHotEncoder。这样,代表 9 月的数字 9 不会被理解为“小于”12 月或 12 月。


推荐阅读