diff options
| -rwxr-xr-x | resources/extract.sh | 2 | ||||
| -rw-r--r-- | src/cnn_train.py (renamed from src/train5.py) | 4 | ||||
| -rw-r--r-- | src/pptdb.c | 11 |
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++]!=','); |