对于淘宝不同种类商品评分进行预测中,使用了torch神经网络模型,但是对于评分低的值预测结果较好,对于评分高的值预测结果不好,波动较大,怎么办
时间: 2023-06-27 10:07:32 浏览: 91
神经网络做预测
这种情况可能是因为评分高的值数量较少,导致模型对这些情况的学习不充分。可以考虑以下几种方法:
1. 数据增强:可以通过数据增强的方式,生成一些评分高的样本,增加模型对这些情况的学习。
2. 调整损失函数:可以尝试使用其他类型的损失函数,如平均绝对误差(MAE)或Huber损失函数,来减小预测结果的波动。
3. 调整模型结构:可以尝试调整模型结构,如增加隐藏层神经元数量、增加层数等,来提高模型对评分高值的拟合能力。
4. 调整超参数:可以尝试调整学习率、正则化参数等超参数,来提高模型的泛化能力,减小预测结果的波动。
阅读全文