An easy to use fully connected neural network library. Also see on Matlab File Exchange.
run the training
modelNN = learnNN(X, y);
plot the confusion matrix for the validation set
plotConfMat(modelNN.confusion_valid);
Here, X
is an [m x n]
feature matrix with m
being the number of examples and n
number of features. y
is an [m x 1]
vector of labels. plotConfMat
plots the confusion matrix for the validation set.
Set some custom options, including the layer structure, regularization parameter lambda
and a choice of activation function.
nnOptions = {'hiddenLayers', [40 20 10], 'lambda', 0.1, 'activationFn', 'tanh'};
Now, run the optimization using the custom options
modelNN = learnNN(X, y, nnOptions);