/ann-framework

Artificial Neural Network framework which allows to build MLP

Primary LanguageJupyter NotebookMIT LicenseMIT

ann-framework

Artificial Neural Network framework which allows to build MLP. It has almost the same interface as Pytorch, more details on this framework and its demonstration with MNIST dataset here.

  • Modules supported by framework: Linear, BatchNormalization, Dropout, ReLU, Sigmoid, Softmax, LogSoftmax,
  • Loss functions: MSELoss, CrossEntropyLoss
  • Optimizers SGD, Adam
  • DataLoader, Sequential