take program flags from command line

tic_tac_toe.py

@@ -4,6 +4,7 @@ from machine_player import MachinePlayer
 from human_player import HumanPlayer
 from player_provider import PlayerProvider
 import json
+import argparse
 
 float_formatter = "{:.3f}".format
 np.set_printoptions(formatter={'float_kind': float_formatter})
@@ -56,48 +57,51 @@ def play(player_provider: PlayerProvider, k_max=10000, with_debug=False):
 				move += 1
 
 
-do_training = 0
+if __name__ == '__main__':
-do_human_player = 0
+	parser = argparse.ArgumentParser()
-do_load_values = 1
+	parser.add_argument("--train", default=False, help="Train before play")
+	parser.add_argument("--num", default=5, help="Number of games to play")
+	parser.add_argument("--human", default=False, help="X-player is human")
+	parser.add_argument("--load", default=True, help="Load experience from previous trainings")
+	args = parser.parse_args()
 
-if do_load_values:
+	if args.load:
-	try:
+		try:
-		with open("values_x.json", "r") as fp:
+			with open("values_x.json", "r") as fp:
-			x_values = json.load(fp)
+				x_values = json.load(fp)
 
-		with open("values_o.json", "r") as fp:
+			with open("values_o.json", "r") as fp:
-			o_values = json.load(fp)
+				o_values = json.load(fp)
-	except Exception:
+		except Exception:
+			x_values = {}
+			o_values = {}
+	else:
 		x_values = {}
 		o_values = {}
-else:
-	x_values = {}
-	o_values = {}
 
-p_hx = HumanPlayer(mark='X', name="Jens")
+	p_mx = MachinePlayer(mark='X', params=MachinePlayer.Params(p_explore=0.1, alpha=0.1), values=x_values)
-p_mx = MachinePlayer(mark='X', params=MachinePlayer.Params(p_explore=0.1, alpha=0.1), values=x_values)
+	p_mo = MachinePlayer(mark='O', params=MachinePlayer.Params(p_explore=0.1, alpha=0.1), values=o_values)
-p_mo = MachinePlayer(mark='O', params=MachinePlayer.Params(p_explore=0.1, alpha=0.1), values=o_values)
 
-K = 5
+	N = args.num
-N = 1000
+	if args.train:
-if do_training:
+		for n in range(0, N):
-	for k in range(0, K):
+			play(PlayerProvider(p_mx, p_mo), 1000, False)
-		play(PlayerProvider(p_mx, p_mo), N, False)
 
-		# Values after training
+			# Values after training
-		# Convert and write JSON object to file
+			# Convert and write JSON object to file
-		with open("values_x.json", "w") as fp:
+			with open("values_x.json", "w") as fp:
-			json.dump(p_mx.values, fp, indent=0)
+				json.dump(p_mx.values, fp, indent=0)
 
-		with open("values_o.json", "w") as fp:
+			with open("values_o.json", "w") as fp:
-			json.dump(p_mo.values, fp, indent=0)
+				json.dump(p_mo.values, fp, indent=0)
 
-		print(f"Iteration: {k*N:8d}")
+			print(f"Iteration: {n*N:8d}")
 
-if do_human_player:
+	if args.human:
-	pp = PlayerProvider(p_hx, p_mo)
+		p_hx = HumanPlayer(mark='X', name="Jens")
-else:
+		pp = PlayerProvider(p_hx, p_mo)
-	pp = PlayerProvider(p_mx, p_mo)
+	else:
+		pp = PlayerProvider(p_mx, p_mo)
 
-play(pp, 10, True)
+	play(pp, 10, True)