def train_new_model(): game_env = PongGame() # init env agent = DqnAgent(game_env) # init agent # input from user print("Enter num of iteration:") while True: try: Dqn.total_num_of_training = int(input()) break except ValueError: print("That's not an integer!") continue # create new file output_directory = JP.create_results_directory() file_name = output_directory + "/data_file.txt" checkpoint_dir = JP.create_checkpoints_directory(output_directory) print(checkpoint_dir) saver = tf.train.Saver() # call train model with tf.Session() as sess: sess.run(tf.global_variables_initializer()) # for initialized Dqn.run_dqn(sess, game_env, agent, output_directory, file_name, saver, checkpoint_dir)