test.py

#!/usr/bin/env python
import matplotlib
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import tasks
import sys
import run_model


def plot(seq_length):
    i, o = tasks.copy(8, seq_length)
    weights, outputs = do_task(i)
    plt.figure(1, figsize=(20, 7))
    plt.subplot(311)
    plt.imshow(i.T, interpolation='nearest')
    plt.subplot(312)
    plt.imshow(o.T, interpolation='nearest')
    plt.subplot(313)
    plt.imshow(outputs.T, interpolation='nearest')
    plt.show()


def plot_weights(seq_length):
    i, o = tasks.copy(8, seq_length)
    weights, outputs = do_task(i)
    plt.figure(1, figsize=(20, 20))
    plt.imshow(weights.T[100:123], interpolation='nearest', cmap=cm.gray)
    plt.show()

if __name__ == "__main__":
    P, do_task = run_model.make_model()
    P.load('l2_low_learning_rate.mdl')
    plot(130)
    # plot_weights(8)