Add support+docs for easy dataset generation
[gostyle.git] / data_about_players.py
blobe70684fb9926924e0c3f2cb33ea1e3b8fd3df88c
1 #!/bin/python
2 from gostyle import PlayerStrategyIdentificator
4 class Data:
5 ### A path to a main pattern file that should contain the most representative patterns along with their occurences
6 ### For its precise format, see `gostyle.py'
7 main_pat_filename = '../pdb/all.pat'
8 ### Folder with player pat files
9 pat_files_folder = '../pdb/pat_files/'
10 ### Pat files should be named ater players
12 ### Thist is used for StrategyOutputVectorGenerator (see gostyle.py)
13 strategy_players = {
14 "territorial":[ 'Cho Chikun', 'Cho U', 'Suzuki Goro'],
15 "agressive":[ 'Kobayashi Koichi', 'Rui Naiwei', 'Yi Se-tol' ],
16 "careful":[ "Yi Ch'ang-ho", 'Ishida Yoshio' ],
17 "moyo":[ 'Takemiya Masaki', 'Cho Tae-hyeon'],
18 "attacking":[ 'Kato Masao', 'Nie Weiping'],
19 "flexible":[ 'Ma Xiaochun' , 'Yasuda Shusaku', 'Kuwahara Shusaku', 'Honinbo Shusaku', 'Go Seigen', 'Shao Zhenzhong' ],
20 "wild":[ 'O Meien', 'Miyazawa Goro', 'Wu Songsheng'],
21 "tesuji focused":[ 'Sakata Eio', 'Luo Xihe', 'Gu Li', 'Jie Li'],
22 "misc":[ 'Yoda Norimoto', 'Chen Zude' ]
25 ### Thist is used for PlanarOutputVectorGenerator (see gostyle.py)
26 player_vector = {
27 "Cho Chikun": [ 1.0, 0.0 ],
28 "Yi Se-tol": [ 0.0, 1.0 ],
29 "Yi Ch'ang-ho old": [ 0.0, -1.0 ],
30 "Ma Xiaochun": [ 0.0, 0.0 ],
31 "Honinbo Shusaku": [ 0.0, 0.0 ],
32 "Go Seigen": [ 0.0, 0.0 ],
33 "Ishida Yoshio": [ 0.0, -0.5 ],
34 "Rui Naiwei": [ 0.5, 1.0 ],
35 "Sakata Eio": [ 0.5, 0.5 ],
36 "Takemiya Masaki": [ -1.0, 0.0 ]}