# HG changeset patch
# User humel
# Date 1272432352 14400
# Node ID 86d5e583e278aeceb44ab99b8a0d315b38475fcb
# Parent  8973abe35a9d887c8b59782519c0d9a276d8fd03
Fixed class number bug

diff -r 8973abe35a9d -r 86d5e583e278 deep/amt/amt.py
--- a/deep/amt/amt.py	Wed Apr 28 00:49:41 2010 -0400
+++ b/deep/amt/amt.py	Wed Apr 28 01:25:52 2010 -0400
@@ -10,6 +10,18 @@
 turks_per_batch = 3
 
 
+def setup_association():
+    answer_assoc = {}
+    for i in range(0,10):
+        answer_assoc[str(i)]=i
+    for i in range(10,36):
+        answer_assoc[chr(i+55)]=i
+    for i in range(36,62):
+        answer_assoc[chr(i+61)]=i
+    return answer_assoc
+
+answer_assoc = setup_association()
+
 def test_error():
     turks = []
     reader = csv.DictReader(open(CVSFILE), delimiter=',')
@@ -43,15 +55,14 @@
     f.close()
     test_error = 0
     for i in range(len(answer_labels)):
-        try: 
-            answer = int(entry[answer_labels[i]])
-        except:
-            try :
-                answer = ord(entry[answer_labels[i]])
+        answer = entry[answer_labels[i]]
+        if len(answer) != 0:
+            try:
+                if answer_assoc[answer] != int(labels[i]):
+                    test_error+=1
             except:
                 test_error+=1
-                continue
-        if answer != int(labels[i]):
+        else:
             test_error+=1
     return test_error