In [5]:

    

import tensorflow as tf
import basic_rnn

def plot_learning_curve(num_steps, state_size=4, epochs=1):
    global losses, total_loss, final_state, train_step, x, y, init_state
    tf.reset_default_graph()
    g = tf.get_default_graph()
    losses, total_loss, final_state, train_step, x, y, init_state = \
        basic_rnn.setup_graph(g,
            basic_rnn.RNN_config(num_steps=num_steps, state_size=state_size))
    res = train_network(epochs, num_steps, state_size=state_size, verbose=False)
    plt.plot(res)


    

In [6]:

    

"""
NUM_STEPS = 1
"""
plot_learning_curve(num_steps=1, state_size=4, epochs=2)


    

    




---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-6-0bcfd4db3966> in <module>()
      2 NUM_STEPS = 1
      3 """
----> 4 plot_learning_curve(num_steps=1, state_size=4, epochs=2)

<ipython-input-5-965decf394f0> in plot_learning_curve(num_steps, state_size, epochs)
      6     tf.reset_default_graph()
      7     g = tf.get_default_graph()
----> 8     losses, total_loss, final_state, train_step, x, y, init_state =         basic_rnn.setup_graph(g,
      9             basic_rnn.RNN_config(num_steps=num_steps, state_size=state_size))
     10     res = train_network(epochs, num_steps, state_size=state_size, verbose=False)

AttributeError: 'module' object has no attribute 'setup_graph'

In [ ]: