diff --git a/main.lua b/main.lua index f4dcef0..7223897 100644 --- a/main.lua +++ b/main.lua @@ -114,6 +114,7 @@ local log_map = { weight_mean = 7, weight_std = 8, test_trial = 9, + decisions = 10, } local function log_csv(t) @@ -307,6 +308,7 @@ local function learn_from_epoch() weight_mean = weight_mean, weight_std = weight_std, test_trial = current_cost or 0, + decisions = decisions_made, } if cfg.enable_network then