danielsabinasz/TensorSlow

Re-implementation of TensorFlow in pure python, with an emphasis on code understandability
Switch branches/tags
Nothing to show
Latest commit 1744c9a Feb 22, 2018
Type Name Latest commit message Commit time
Failed to load latest commit information.
examples Aug 24, 2017
jupyter_notebook Feb 22, 2018
tensorslow Aug 24, 2017
README.md Aug 25, 2017

TensorSlow

A re-implementation of TensorFlow functionality in pure python

TensorSlow is a minimalist machine learning API that mimicks the TensorFlow API, but is implemented in pure python (without a C backend). The source code has been built with maximal understandability in mind, rather than maximal efficiency. Therefore, TensorSlow should be used solely for educational purposes. If you want to understand how deep learning libraries like TensorFlow work under the hood, this may be your best shot.

I have written an article in my blog at deepideas.net that develops this library step by step, explaining all the math and algorithms on the way: Deep Learning From Scratch.

How to use

Import:

``````import tensorslow as ts
``````

Create a computational graph:

``````ts.Graph().as_default()
``````

Create input placeholders:

``````training_features = ts.placeholder()
training_classes = ts.placeholder()
``````

Build a model:

``````weights = ts.Variable(np.random.randn(2, 2))
biases = ts.Variable(np.random.randn(2))
model = ts.softmax(ts.add(ts.matmul(X, W), b))
``````

Create training criterion:

``````loss = ts.negative(ts.reduce_sum(ts.reduce_sum(ts.multiply(training_classes, ts.log(model)), axis=1)))
``````

Create optimizer:

``````optimizer = ts.train.GradientDescentOptimizer(learning_rate=0.01).minimize(J)
``````

Create placeholder inputs:

``````feed_dict = {
training_features: my_training_features,
training_classes: my_training_classes
}
``````

Create session:

``````session = ts.Session()
``````

Train:

``````for step in range(100):
loss_value = session.run(loss, feed_dict)
if step % 10 == 0:
print("Step:", step, " Loss:", loss_value)
session.run(optimizer, feed_dict)
``````

Retrieve model parameters:

``````weights_value = session.run(weigths)
biases_value = session.run(biases)
``````

Check out the `examples` directory for more.