First commit

plotboardacc.py

@@ -0,0 +1,46 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*- 
+
+__author__ = "Aleksey Lobanov"
+__copyright__ = "Copyright 2016, Aleksey Lobanov"
+__credits__ = ["Aleksey Lobanov"]
+__license__ = "MIT"
+__maintainer__ = "Aleksey Lobanov"
+__email__ = "i@likemath.ru"
+
+import matplotlib.pyplot as plt
+import matplotlib.cm as cm
+import matplotlib.patches as mpatches
+
+from neurallife import loadKeras, generateData, getAccuracies, META_PARAMETERS
+
+
+if __name__ == "__main__":
+    plt.title("Neural Life")
+
+    plt.xscale("log")
+    plt.yscale("log")
+    
+    plt.xlabel("Train size")
+    plt.ylabel("Board accuracy")
+
+    plt_patches = []
+    for meta_ind,N in enumerate(sorted(META_PARAMETERS.keys())):
+        points_x = []
+        points_y = []
+        
+        X_test, Y_test = generateData(N, 100000)
+        
+        for data_size in META_PARAMETERS[N]: 
+            nn = loadKeras("models/model_{}_{}".format(N, int(data_size * 0.5)))
+        
+            cellAcc, boardAcc = getAccuracies(nn, X_test, Y_test)
+            
+            points_x.append(data_size * 0.5)
+            points_y.append(boardAcc)
+            
+        plt.plot(points_x, points_y, "o", linestyle="-", color=cm.ocean(meta_ind/len(META_PARAMETERS)))
+        plt_patches.append(mpatches.Patch(color=cm.ocean(meta_ind/len(META_PARAMETERS)), label="N={}".format(N)))
+
+    plt.legend(handles=plt_patches, loc=2)
+    plt.savefig("graphics-board.svg")