deep-learning-scala
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:
- Mini-library for subset of Tensor calculus
- Mini-library for data preparation
- A DSL for Neural Network creation, including layers
- Pluggable weights optimizer and initializer
- Pluggable implementation of activation and loss functions
- Pluggable training metric calculation
See examples:
- Classification with Artificial Neural Network
- Linear Regression
- MNIST Images Classification