diff options
Diffstat (limited to 'src/nn_train.py')
| -rw-r--r-- | src/nn_train.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/nn_train.py b/src/nn_train.py index 12ce868..f2550b4 100644 --- a/src/nn_train.py +++ b/src/nn_train.py @@ -56,10 +56,10 @@ def train(size, model, data, iterations=1, epochs=10, batch=None): tra_res = model.evaluate(tra_input, tra_outcome, verbose=False) results.append((tra_res, val_res)) print(val_res) - write_weights(model, i+1, (tra_res, val_res)) + write_weights(model, str(i+1).zfill(len(str(iterations))), (tra_res, val_res)) print("\nScores") for i, data in enumerate(results): - print(f"Iteration {i}: {data}") + print(f"Iteration {i+1}: {data}") return results @@ -84,12 +84,12 @@ def write_weights(model, iteration, performance): output_weights = transpose(model.trainable_variables[2], perm=[1, 0]) output_bias = model.trainable_variables[3] # Prepare output - to_output = [("dense1_weights[DENSE_NUM][INP_NUM]",dense1_weights) + to_output = [("dense1_weights[DENSE_NUM][INP_NUM]", dense1_weights), ("dense1_biases[DENSE_NUM]", dense1_biases), ("output_weights[2][DENSE_NUM]", output_weights), ("output_bias[2]", output_bias)] # Write to file - f = open("weights-"+str(iteration)+".txt", "w") + f = open("weights-"+iteration+".txt", "w") f.write("/*\n") model.summary(print_fn=lambda l: f.write(" * "+l+"\n")) f.write(" * "+str(performance)+"\n*/\n\n") @@ -103,4 +103,4 @@ data = load_data(5) model = make_model(5, 64) print("Before training", model.evaluate(data[1][0], data[1][1], verbose=False, batch_size=16)) -results = train(5, model, data, iterations=1, epochs=10, batch=None) +results = train(5, model, data, iterations=5, epochs=10, batch=128) |