graphAttack

Build Status

​GraphAttack is a simple computational graph library for machine learning in pure Python (you can think of simple tensorflow). The main point is to combine mathematical operation together to form a workflow of choice in the form of a graph. The graph then takes care of pushing the data forward and propagating the gradients backwards.

I have aimed for the library to be simple and transparent so that it would be easy to understand and modify to fit individual needs. The performance was not the primary objective as there are plenty of fast alternatives; the aim was smaller, educational models.

All feedback and ideas for improvement welcome.

Tutorial

Examples

Setting up and running / training a dense neural network model

Setting up and running / training a Convolution neural network model

Setting up and running / training a Recurrent neural network

Features

Matrix operations

Regularization

Activations

Convolution

Cost operations

Optimisation

Misc

GPU Support

Limited GPU support can be made avaliable by modifying the file: graphAttack/gaUtilities/graphAttackFunctions.py

Commenting out lines 1-2 and uncommenting lines 4-22 will send the dot product computations to your GPU

Requirements

Additional Resources

http://www.deeplearningbook.org/ section 6.5.1 for more information on computational graphs and the rest of the book for more details about ML/deep learning.

Dependencies

License

MIT