Closed stevenydc closed 9 years ago
You want Output_num_units=2
. y
should be one dimensional and have shape (8,)
.
I've put together a little script with toy examples that demonstrates what kind of data is expected for classification and regression problems.
Perhaps give an example of how the user could one-hot encode the values of y him self? Right now nolearn, etc chokes if the y
passed in for classification is one-hot encoded already.
Related #10.
I have a binary classification problem... Initially I had :
The network worked fine (as in no error) But then I realized that it doesn't make sense to have regression=True for a classification problem. So set regression = False.... but that gives me error when I try to train the net:
So I forced the shape to be (8,1) by doing trainY = trainY.reshape(8,1)
This allowed me to train the network without error _IF_ eval_size = 0 if I set eval_size to 0.2, then I get the following error:
And I can't seem to resolve this problem... please help!