From 475df1775f91b6f89f4232a030288d9ee575c7de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=92=B2=E6=BA=90?= <2402552459@qq.com> Date: Sun, 13 Aug 2023 19:46:52 +0800 Subject: [PATCH] polish(pu): polish get_katago_statistics_for_dataset --- zoo/board_games/go/envs/get_katago_statistics_for_dataset.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/zoo/board_games/go/envs/get_katago_statistics_for_dataset.py b/zoo/board_games/go/envs/get_katago_statistics_for_dataset.py index 61f4a7df0..714413ce4 100644 --- a/zoo/board_games/go/envs/get_katago_statistics_for_dataset.py +++ b/zoo/board_games/go/envs/get_katago_statistics_for_dataset.py @@ -1,7 +1,7 @@ import numpy as np from easydict import EasyDict -from get_episode_gtp_actions import read_episode_gtp_actions_from_txt +from read_episode_gtp_actions import read_episode_gtp_actions_from_txt from zoo.board_games.go.envs.go_env import GoEnv from zoo.board_games.go.envs.katago_policy import KatagoPolicy @@ -43,6 +43,8 @@ def get_katago_statistics_for_dataset(): - katago_statistics_step (:obj:`dict`): The katago statistics for each step. Note: + katago policy ckpt is download form https://media.katagotraining.org/uploaded/networks/zips/kata1/kata1-b18c384nbt-s6582191360-d3422816034.zip + we should unzip it to ./code/KataGo/kata1-b18c384nbt-s6582191360-d3422816034. - katago_statistics_step.keys() (:obj:`list`): The keys of katago_statistics_step. - dict_keys(['policy0', 'policy1', 'moves_and_probs0', 'moves_and_probs1', 'value', 'td_value', 'td_value2', 'td_value3', 'scoremean', 'td_score', 'scorestdev', 'lead', 'vtime', 'estv', 'ests', 'ownership', 'ownership_by_loc', 'scoring', 'scoring_by_loc', 'futurepos', 'futurepos0_by_loc', 'futurepos1_by_loc', 'seki', 'seki_by_loc', 'seki2', 'seki_by_loc2', 'scorebelief', 'genmove_result']) - katago_statistics_step['moves_and_probs0'] (:obj:`list`): The moves and probs for player 0.