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)
