Skip to content

A fully connected neural network classifier with arbitrary number of hidden layers, different activation functions, etc..

License

Notifications You must be signed in to change notification settings

vtshitoyan/simpleNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

simpleNN

An easy to use fully connected neural network library. Also see on Matlab File Exchange.

Example usages

Basic

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.

Custom

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);

About

A fully connected neural network classifier with arbitrary number of hidden layers, different activation functions, etc..

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages