Syntax errors, small tweak to training data generation

3 files changed, 9 insertions, 8 deletions

diff --git a/resources/extract.sh b/resources/extract.sh
index fb6f267..7fdda28 100755
--- a/resources/extract.sh
+++ b/resources/extract.sh
@@ -62,7 +62,7 @@ fi
 
 make pptdb
 
-for size in 5; do
+for size in 5 6; do
 		echo
 		extract $size notation,result
 		process $size 400000 training
diff --git a/src/train5.py b/src/cnn_train.py
index 25b1aab..0580b03 100644
--- a/src/train5.py
+++ b/src/cnn_train.py
@@ -155,9 +155,9 @@ def write_weights(model):
     conv2d_weights = transpose(model.trainable_variables[0], perm=[3, 0, 1, 2])
     conv2d_biases = model.trainable_variables[1]
     dense1_weights = transpose(model.trainable_variables[2], perm=[1, 0])
-    dense1_biases = model.trainable_variables[3])
+    dense1_biases = model.trainable_variables[3]
     dense2_weights = transpose(model.trainable_variables[4], perm=[1, 0])
-    dense2_biases = model.trainable_variables[5])
+    dense2_biases = model.trainable_variables[5]
     output_weights = transpose(model.trainable_variables[6], perm=[1, 0])[0]
     output_bias = model.trainable_variables[7][0]
     # Prepare formatting
diff --git a/src/pptdb.c b/src/pptdb.c
index b5bdfa4..ecf5caa 100644
--- a/src/pptdb.c
+++ b/src/pptdb.c
@@ -133,12 +133,13 @@ parse_line(const char *pt, const ssize_t read) {
 				}
 			}
 		}
-		// Generate training data, not too early in the game, random
-		// orientation
+		// Generate training data, not too early in the game, all
+		// orientations
 		if (generate && ply + 4 >= total_plies) {
-			int dx = (rand()&1)?-1:+1;
-			int dy = (rand()&1)?-1:+1;
-			write_input(dx,dy);
+			write_input(-1,-1);
+			write_input(-1,+1);
+			write_input(+1,-1);
+			write_input(+1,+1);
 		}
 		// Parse next action
 		while (idx<read && pt[idx++]!=',');