我正在尝试使用 deepnet 包的 dbn.dnn.train 函数训练 MNIST 数据集。该任务是一个分类任务。
我正在使用以下命令:
dbn.deepnet <- dbn.dnn.train(train.image.data,train.image.labels,hidden=c(5,5))
我面临的问题是:
1)标签应该是因子类型向量。但是,当我将标签输入为因子时,函数会出现错误,指出“y应该是矩阵或向量”。因此,我将标签用作数字。如何进行分类任务?
2)用于对dbn.dnn.train进行预测的函数是什么?我正在使用nn.predict,但文档提到输入应该是由函数nn.train训练的神经网络(dbn.dnn.train未被提及)。所有记录的输出为0.9986。
nn.predict(dbn.deepnet,train.image.data)
deepnet
训练MNIST的示例,加速信息在这里。 - Patric