novakov-alexey-zz/deep-learning-scala

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