Use namedtuple for clarity.

Marina Samuel · Marina Samuel · commit 35719ba00bb2 · 2014-05-19T21:20:38.000-04:00
diff --git a/ocr/ocr.py b/ocr/ocr.py
@@ -4,6 +4,7 @@
 import numpy as np
 from numpy import matrix
 from math import pow
+from collections import namedtuple
 import math
 import random
 import os
@@ -38,7 +39,8 @@ def __init__(self, num_hidden_nodes, data_matrix, data_labels, training_indices,
             self.hidden_layer_bias = self._rand_initialize_weights(1, 10)
 
             # Train using sample data
-            self.train([{"y0":self.data_matrix[i], "label":int(self.data_labels[i])} for i in training_indices])
+            TrainData = namedtuple('TrainData', ['y0', 'label'])
+            self.train([TrainData(self.data_matrix[i], int(self.data_labels[i])) for i in training_indices])
             self.save()
         else:
             self._load()
