def test_model(file):
args = initial_args
print(f'File: {file}')
print('################################################################################')
result = []
args.checkpoint = './temp_models/'
net = nn(args)
if file != 'none':
net.load_checkpoint(args.checkpoint, filename=file)
else:
net = None
for sim in num_simulations:
print("testing simulations", sim)
wins = 0
for i in range(n_games):
predicted_cost = predict_path(games[i], sim, args, net)
if optimal[i] * error >= predicted_cost:
wins += 1
result.append(wins / n_games)
# print(result)
#print(result)
return (file, result)
def run_simulation(arguments):
sim, args, file = arguments
args.numMCTSSims = sim
wins = 0
# Load net
net = nn(args)
if file != 'none':
net.load_checkpoint(args.checkpoint, filename=file)
else:
net = None
# Simulate n games
for i in range(n_games):
predicted_cost = predict_path(games[i], sim, args, net)
if optimal[i] * error >= predicted_cost:
wins += 1
return (sim, wins / n_games)
def last_test_model(file):
print(f'File: {file}')
global global_args
global global_net
global global_sim
global_args = initial_args
global_args.checkpoint = './temp_models/'
global_net = nn(global_args)
result = []
if file != 'none':
global_net.load_checkpoint(global_args.checkpoint, filename=file)
else:
global_net = None
for sim in num_simulations:
#print("testing simulations", sim)
global_sim = sim
def last_test_model(file):
print(f'File: {file}')
global global_args
global global_net
global global_sim
global_args = initial_args
global_args.checkpoint = './temp_models/'
global_net = nn(global_args)
result = []
if file != 'none':
global_net.load_checkpoint(global_args.checkpoint, filename=file)
else:
global_net = None
for sim in num_simulations:
#print("testing simulations", sim)
global_sim = sim
pool = Pool()
game_results = pool.map(run_game, list(range(n_games)))
pool.close()
pool.join()
result.append(sum(game_results) / n_games)
#print(result)
return result
'n_executions': 100,
'lr': 0.001,
'dropout': 0.3,
'epochs': 10,
'batch_size': 64,
'use_gdc': True
})
#sims = [5, 25, 50, 75, 100, 150, 200, 250, 300, 400, 500, 600, 700, 800, 900, 1000]
#sims = [250]
#sims = [250, 800, 900, 1000]
n_games = 300
error = 1.4
net = nn(args)
#ims = [5, 25, 50]
files = [
'best_model_25_mctsSims.pth.tar', 'best_model_50_mctsSims.pth.tar',
'best_model_5_mctsSims.pth.tar', 'checkpoint_5sims_1.pth.tar',
'best_model_75_mctsSims.pth.tar', 'checkpoint_25sims_3.pth.tar',
'checkpoint_50sims_1.pth.tar'
]
for file in files:
#file = f'best_model_{sim}_mctsSims.pth.tar'
args.checkpoint = './new_models/'
#file = 'best.pth.tar'
net.load_checkpoint(args.checkpoint, filename=file)
wins = 0