首页 > 解决方案 > Mxnet 预测训练使用 rmse 但有些问题

问题描述

当我进行预测训练时,我想RMSE用作我的损失函数。但是当我使用它时,它总是返回numpy.float64. 而且我想做梯度下降,所以我不知道如何转换数据类型。你知道,只能NDarray做梯度下降。谢谢大家!

在此处输入图像描述

在此处输入图像描述

标签: predictionmxnet

解决方案


推荐阅读