def train(specified_network=0,
specified_model_name="",
round_times=0,
allow_user_input=True):
network = select_network(is_training=True,
specified_network=specified_network,
specified_model_name=specified_model_name)
if isinstance(network, PolicyValueNet_from_junxiaosong):
train_with_net_junxiaosong(network, allow_user_input, round_times)
elif isinstance(network, PolicyValueNet_ResNet):
train_with_resnet(network, allow_user_input, round_times)
def player_init(player_selected, name):
if player_selected == 1:
return Human(name=name)
elif player_selected == 2:
search_times, greedy_value = set_AI_conf(search_times=2000, greedy_value=5.0)
return AI_MCTS(name=name,
search_times=search_times,
greedy_value=greedy_value,
is_output_analysis=conf.conf_dict["AI_is_output_analysis"])
elif player_selected == 3:
network = select_network()
search_times, greedy_value = set_AI_conf(search_times=400, greedy_value=5.0)
return AI_MCTS_Net(name=name,
policy_value_function=network.predict,
board_to_xlabel=network.board_to_xlabel,
is_training=False,
search_times=search_times,
greedy_value=greedy_value,
is_output_analysis=conf.conf_dict["AI_is_output_analysis"])