Misc

babi_runner.py

@@ -14,6 +14,7 @@
 random.seed(seed_val)
 np.random.seed(seed_val)  # for reproducing
 
+
 def run_task(data_dir, task_id):
     """
     Train and test for each task

demo/qa.py

@@ -173,6 +173,7 @@ def train_model(data_dir, model_file):
     memn2n = MemN2N(data_dir, model_file)
     memn2n.train()
 
+
 def run_console_demo(data_dir, model_file):
     """
     Console-based demo
@@ -231,6 +232,7 @@ def run_console_demo(data_dir, model_file):
         if will_continue != '' and will_continue.lower() != 'y': break
         print("=" * 70)
 
+
 def run_web_demo(data_dir, model_file):
     from demo.web import webapp
     webapp.init(data_dir, model_file)

demo/web/webapp.py

@@ -12,6 +12,7 @@
 memn2n = None
 test_story, test_questions, test_qstory = None, None, None
 
+
 def init(data_dir, model_file):
     """ Initialize web app """
     global memn2n, test_story, test_questions, test_qstory
@@ -26,13 +27,16 @@ def init(data_dir, model_file):
     test_story, test_questions, test_qstory = \
         parse_babi_task(test_data_path, memn2n.general_config.dictionary, False)
 
+
 def run():
     app.run()
 
+
 @app.route('/')
 def index():
     return flask.render_template("index.html")
 
+
 @app.route('/get/story', methods=['GET'])
 def get_story():
     question_idx      = np.random.randint(test_questions.shape[1])
@@ -52,6 +56,7 @@ def get_story():
         "correct_answer": correct_answer
     })
 
+
 @app.route('/get/answer', methods=['GET'])
 def get_answer():
     question_idx  = flask.request.args.get('question_idx')

memn2n/memory.py

@@ -3,6 +3,7 @@
 from memn2n.nn import ElemMult, Identity, Sequential, LookupTable, Module
 from memn2n.nn import Sum, Parallel, Softmax, MatVecProd
 
+
 class Memory(Module):
     """
     Memory:
@@ -119,6 +120,7 @@ def init_output_module(self):
         self.mod_out.add(p)
         self.mod_out.add(MatVecProd(False))
 
+
 class MemoryL(Memory):
     """
     MemoryL:

memn2n/nn.py

@@ -6,6 +6,7 @@
 # when Softmax is not included at the end layer.
 np.seterr(divide='ignore')
 
+
 class Module(object):
     """
     Abstract Module class for neural net

train_test.py

@@ -164,6 +164,7 @@ def train_linear_start(train_story, train_questions, train_qstory, memory, model
     # Train with old settings
     train(train_story, train_questions, train_qstory, memory, model, loss, general_config)
 
+
 def test(test_story, test_questions, test_qstory, memory, model, loss, general_config):
     total_test_err = 0.
     total_test_num = 0