Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore
LICENSE
README.md
ann.go
data.go
main.go
mlp.go
sgd.go
sgd_test.go
validate.go

README.md

Gotrain fits feedforward backpropigation neural networks to classification data.

Usage

Create a toy dataset and train a multilayer perceptron network (MLP):

$ cat > ./xor.csv <<EOF
1,0,1
1,1,0
0,0,0
0,1,1
EOF
$ ./gotrain -datapath ./xor.csv MLP

For help on global flags and available artificial neuro networks (ANNs):

$ ./gotrain -h

For help on model specific flags:

$ ./gotrain MLP -h

Future enhancements

  • Support model persistence in PMML.
  • Describe network topology outside of code.
  • Mutate network topology during training to minimize neuron count.
  • Optimize hyperparameters during training to converge faster.
  • Implement more training algorithms.
You can’t perform that action at this time.