├── .gitignore ├── Makefile ├── README.md ├── configs ├── .gitkeep ├── mlm_base.yaml └── policy_value_base.yaml ├── docker ├── Dockerfile └── entrypoint.sh ├── engine ├── mcts_player.sh └── policy_player.sh ├── env_name.yml ├── requirements.txt ├── setup.py ├── src ├── data │ ├── __init__.py │ ├── mlm.py │ └── policy_value.py ├── features │ ├── __init__.py │ ├── common.py │ └── policy_value.py ├── model │ ├── __init__.py │ └── bert.py ├── pl_modules │ ├── __init__.py │ ├── mlm.py │ └── policy_value.py ├── player │ ├── __init__.py │ ├── base_player.py │ ├── mcts_player.py │ ├── policy_player.py │ └── usi.py ├── uct │ ├── __init__.py │ └── uct_node.py └── utils │ ├── __init__.py │ ├── hcpe.py │ ├── misc.py │ ├── sfen.py │ └── shogi.py └── tools ├── download_and_build_lesserkai.sh ├── make_dataset.py ├── pl_to_transformers.py ├── test_engine.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/.gitignore -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/README.md -------------------------------------------------------------------------------- /configs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/mlm_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/configs/mlm_base.yaml -------------------------------------------------------------------------------- /configs/policy_value_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/configs/policy_value_base.yaml -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/entrypoint.sh: -------------------------------------------------------------------------------- 1 | python setup.py develop 2 | exec "$@" 3 | -------------------------------------------------------------------------------- /engine/mcts_player.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/engine/mcts_player.sh -------------------------------------------------------------------------------- /engine/policy_player.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/engine/policy_player.sh -------------------------------------------------------------------------------- /env_name.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/env_name.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/setup.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/data/mlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/data/mlm.py -------------------------------------------------------------------------------- /src/data/policy_value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/data/policy_value.py -------------------------------------------------------------------------------- /src/features/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/features/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/features/common.py -------------------------------------------------------------------------------- /src/features/policy_value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/features/policy_value.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/model/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/model/bert.py -------------------------------------------------------------------------------- /src/pl_modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/pl_modules/__init__.py -------------------------------------------------------------------------------- /src/pl_modules/mlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/pl_modules/mlm.py -------------------------------------------------------------------------------- /src/pl_modules/policy_value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/pl_modules/policy_value.py -------------------------------------------------------------------------------- /src/player/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/player/base_player.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/player/base_player.py -------------------------------------------------------------------------------- /src/player/mcts_player.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/player/mcts_player.py -------------------------------------------------------------------------------- /src/player/policy_player.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/player/policy_player.py -------------------------------------------------------------------------------- /src/player/usi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/player/usi.py -------------------------------------------------------------------------------- /src/uct/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/uct/uct_node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/uct/uct_node.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/hcpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/utils/hcpe.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/sfen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/utils/sfen.py -------------------------------------------------------------------------------- /src/utils/shogi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/src/utils/shogi.py -------------------------------------------------------------------------------- /tools/download_and_build_lesserkai.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/tools/download_and_build_lesserkai.sh -------------------------------------------------------------------------------- /tools/make_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/tools/make_dataset.py -------------------------------------------------------------------------------- /tools/pl_to_transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/tools/pl_to_transformers.py -------------------------------------------------------------------------------- /tools/test_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/tools/test_engine.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nyoki-mtl/bert-mcts-youtube/HEAD/tools/train.py --------------------------------------------------------------------------------