/ Machine Learning

A lightweight deep learning framework written in Python3

A lightweight deep learning framework written in Python3

tinynn

tinynn is a lightweight deep learning framework written in Python3 (with NumPy).

Getting Started

Install

pip install tinynn

Examples

git clone https://github.com/borgwang/tinynn.git
cd tinynn/examples

# MNIST classification
python mnist/run.py  

# a toy regression task
python nn_paint/run.py  

# reinforcement learning demo (gym environment required)
python rl/run.py

Components

  • layers: Dense, Conv2D, ConvTranspose2D, RNN, MaxPool2D, Dropout, BatchNormalization
  • activation: ReLU, LeakyReLU, Sigmoid, Tanh, Softplus
  • losses: SoftmaxCrossEntropy, SigmoidCrossEntropy, MAE, MSE, Huber
  • optimizer: RAdam, Adam, SGD, Momentum, RMSProp, Adagrad, Adadelta

GitHub