deep-learning-scala icon indicating copy to clipboard operation
deep-learning-scala copied to clipboard

Artificial Neural Network in Scala

Deep Learning in Scala from scratch

Example of mini-library implementation for Artificial Neural Networks training and inference.

Implementation includes:

  1. Mini-library for subset of Tensor calculus
  2. Mini-library for data preparation
  3. A DSL for Neural Network creation, including layers
  4. Pluggable weights optimizer and initializer
  5. Pluggable implementation of activation and loss functions
  6. Pluggable training metric calculation

See examples:

  • Classification with Artificial Neural Network
  • Linear Regression
  • MNIST Images Classification