错误消息学习简单模型mxnet

时间:2017-10-26 20:53:43

标签: mxnet

我收到此错误消息

  

mx.model.select.layout.train(X,y)出错:无法自动选择   array.layout,请指定此参数

我正在尝试创建简单的模型以了解其工作原理

创建数据

train.x <- data.matrix(sample(1:100,1000,replace=T) )
colnames(train.x) <- "X"
train.y <- data.matrix(train.x^2)
colnames(train.y) <- "Y"

test.x = data.matrix(sample(1:100,50,replace=T) )
colnames(test.x) <- "X"
test.y <-data.matrix(test.x^2)
colnames(test.y) <- "Y"

测试数据

mx.set.seed(0)
model <- mx.mlp(train.x, train.y, hidden_node=10, out_node=2, 
out_activation="softmax",
num.round=20, array.batch.size=15, learning.rate=0.07, momentum=0.9,
eval.metric=mx.metric.accuracy)

1 个答案:

答案 0 :(得分:1)

看起来你正试图学习函数x的近似值 - &gt; x ^ 2.您的激活功能不应该是“softmax”。这对分类问题更合适。您可以使MSE(均方误差)或其他更适合回归问题的损失函数。

您可能还会发现此MXNet / R教程很有帮助: https://mxnet.incubator.apache.org/tutorials/r/fiveMinutesNeuralNetwork.html