我收到此错误消息
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 回答
看起来你正在尝试学习函数x - > x ^ 2的近似值 . 你的激活函数不应该是“softmax” . 这对分类问题更合适 . 您可以使MSE(均方误差)或其他更适合回归问题的损失函数 .
您可能还会发现此MXNet / R教程很有用:https://mxnet.incubator.apache.org/tutorials/r/fiveMinutesNeuralNetwork.html