Description
I have a question about the MNIST sample code.
https://github.com/RubixML/MNIST
In the MNIST example on git, you say:
The output layer adds an additional layer of neurons with a Softmax activation making this particular network architecture 4 layers deep.
$estimator = new PersistentModel(
new Pipeline([
new ImageResizer(28, 28),
new ImageVectorizer(true),
new ZScaleStandardizer(),
], new MultiLayerPerceptron([
new Dense(100),
new Activation(new LeakyReLU()),
new Dropout(0.2),
new Dense(100),
new Activation(new LeakyReLU()),
new Dropout(0.2),
new Dense(100),
new Activation(new LeakyReLU()),
new Dropout(0.2),
], 256, new Adam(0.0001))),
new Filesystem('mnist.rbx', true)
);
However, in the sample code, I can't see any settings for the output layer. In the sample code, I can only see the settings for three hidden layers. Where are the settings for the output layer?