supervised trainer saves all logs to metadata and is safe if no validation data

wrongu · wrongu · commit e52a1afb6d83 · 2016-03-30T09:37:19.000-04:00
diff --git a/AlphaGo/training/supervised_policy_trainer.py b/AlphaGo/training/supervised_policy_trainer.py
@@ -50,18 +50,21 @@ def __init__(self, path):
 		}
 
 	def on_epoch_end(self, epoch, logs={}):
-		self.metadata["epochs"].append({
-			"acc": logs.get("acc"),
-			"val_acc": logs.get("val_acc")
-		})
+		self.metadata["epochs"].append(logs)
 
-		best_accuracy = self.metadata["epochs"][self.metadata["best_epoch"]]["val_acc"]
-		if logs.get("val_acc") > best_accuracy:
+		if "val_loss" in logs:
+			key = "val_loss"
+		else:
+			key = "loss"
+
+		best_loss = self.metadata["epochs"][self.metadata["best_epoch"]][key]
+		if logs.get(key) < best_loss:
 			self.metadata["best_epoch"] = epoch
 
 		with open(self.file, "w") as f:
 			json.dump(self.metadata, f)
 
+
 BOARD_TRANSFORMATIONS = [
 	lambda feature: feature,
 	lambda feature: np.rot90(feature, 1),
