jeremiedb commented on issue #9967: Error in output.shape[[output.names]] URL: https://github.com/apache/incubator-mxnet/issues/9967#issuecomment-371073140 `mx.model.FeedForward.create` is a function intended for training of the model, it is therefore expected to have a label passed as the `y` argument. Your model should have a loss function for in order to train; for example, `mx.symbol.LinearRegressionOutput`. Please take a look at this MNIST tutorial for example: http://mxnet.incubator.apache.org/tutorials/r/mnistCompetition.html. You need to be aware of the the data dimensions through the model: if 126 samples in X, Y should also be of length 126. Check the model arhcitecture used: a FC on images (which should normally be of equal WH dimensions) is likely not what is needed (check for convolution layers). And if performing a 2 class classification, then needs a final FC with num_hidden = 2 followed by a softmaxoutput or rather a FC of size 1 followed by a LogisticLinearOutput.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
