Merge branch 'master' of ssh://repo.or.cz/srv/git/gostyle
[gostyle.git] / data_about_players.py
blob9c124c1b6aa905b836f091e6243b52d926d848c9
1 #!/usr/bin/python
2 from gostyle import PlayerStrategyIdentificator
4 # TODO following method is total XXX
5 # basically whole concept of static class Data is somewhat shallow, and not a good style Python
6 def questionare_average(questionare_list,silent=False):
7 import numpy
8 total={}
9 count={}
10 for questionare in questionare_list:
11 for name in questionare.keys():
12 if name in count:
13 count[name] += 1
14 else:
15 count[name] = 1
16 total[name] = []
17 total[name].append(questionare[name])
19 l=[]
20 for name, counter in count.items():
21 l.append( (counter, name) )
22 l.sort()
23 l.reverse()
24 variance=[]
25 result={}
26 for counter, name in l:
27 if counter > 1:
28 means=[]
29 if not silent:
30 space = 0 if len(name) > 20 else ' '*(20 - len(name))
31 print "%d: %s"%(counter, name + space),
32 a = numpy.array(total[name]).transpose()
33 for b in a:
34 means.append(b.mean())
35 if not silent:
36 print u"%2.3f \u00B1 %2.3f "%(b.mean(), numpy.sqrt(b.var())),
37 variance.append(numpy.sqrt(b.var()))
38 if not silent:
39 print
40 variance.append(numpy.sqrt(b.var()))
41 result[name] = means
42 if not silent:
43 print "Mean standard deviation is: %2.3f"%(numpy.array(variance).mean(),)
44 return result
46 class Data:
47 ### A path to a main pattern file that should contain the most representative patterns along with their occurences
48 ### For its precise format, see `gostyle.py'
49 main_pat_filename = '../pdb-cont/all.pat'
50 ### Folder with player pat files
51 pat_files_folder = '../pdb-cont/pat_files/'
52 ### Pat files should be named ater players
54 ### Explicit list of players
55 players_all = ['Takemiya Masaki', 'Cho Tae-hyeon', 'Yoda Norimoto', 'Chen Zude', 'Sakata Eio', 'Luo Xihe', 'Gu Li', 'Jie Li', 'Cho Chikun', 'Cho U', 'Suzuki Goro', 'O Meien', 'Miyazawa Goro', 'Wu Songsheng', 'Ma Xiaochun', 'Yasuda Shusaku', 'Kuwahara Shusaku', 'Honinbo Shusaku', 'Go Seigen', 'Shao Zhenzhong', "Yi Ch'ang-ho", 'Ishida Yoshio', 'Kobayashi Koichi', 'Rui Naiwei', 'Yi Se-tol', 'Kato Masao', 'Nie Weiping']
56 #players_all = ['Chen Yaoye', 'Chen Zude', 'Cho Chikun', 'Cho Tae-hyeon', 'Cho U', 'Fujisawa Hideyuki', 'Go Seigen', 'Gu Li', 'Hane Naoki', 'Honinbo Dosaku', 'Honinbo Shusaku', 'Honinbo Shuwa', 'Huang Longshi', 'Ishida Yoshio', 'Jie Li', 'Kato Masao', 'Kato Shin', 'Kobayashi Koichi', 'Kuwahara Shusaku', 'Luo Xihe', 'Ma Xiaochun', 'Miyazawa Goro', 'Nie Weiping', 'O Meien', 'Otake Hideo', 'Rui Naiwei', 'Sakata Eio', 'Shao Zhenzhong', 'Suzuki Goro', 'Takao Shinji', 'Takemiya Masaki', 'Wu Songsheng', 'Yamashita Keigo', 'Yasuda Shusaku', "Yi Ch'ang-ho", 'Yi Se-tol', 'Yoda Norimoto', 'Yuki Satoshi']
58 ### Thist is used for StrategyOutputVectorGenerator (see gostyle.py)
59 strategy_players = {
60 "territorial":[ 'Cho Chikun', 'Cho U', 'Suzuki Goro'],
61 "agressive":[ 'Kobayashi Koichi', 'Rui Naiwei', 'Yi Se-tol' ],
62 "careful":[ "Yi Ch'ang-ho", 'Ishida Yoshio' ],
63 "moyo":[ 'Takemiya Masaki', 'Cho Tae-hyeon'],
64 "attacking":[ 'Kato Masao', 'Nie Weiping'],
65 "flexible":[ 'Ma Xiaochun' , 'Yasuda Shusaku', 'Kuwahara Shusaku', 'Honinbo Shusaku', 'Go Seigen', 'Shao Zhenzhong' ],
66 "wild":[ 'O Meien', 'Miyazawa Goro', 'Wu Songsheng'],
67 "tesuji focused":[ 'Sakata Eio', 'Luo Xihe', 'Gu Li', 'Jie Li'],
68 "misc":[ 'Yoda Norimoto', 'Chen Zude' ]
71 ### Thist is used for PlanarOutputVectorGenerator (see gostyle.py)
72 player_vector = {
73 "Cho Chikun": [ 1.0, 0.0 ],
74 "Yi Se-tol": [ 0.0, 1.0 ],
75 "Yi Ch'ang-ho": [ 0.0, -1.0 ],
76 "Ma Xiaochun": [ 0.0, 0.0 ],
77 #"Honinbo Shusaku": [ 0.0, 0.0 ],
78 #"Go Seigen": [ 0.0, 0.0 ],
79 "Ishida Yoshio": [ 0.0, -0.5 ],
80 "Rui Naiwei": [ 0.5, 1.0 ],
81 "Sakata Eio": [ 0.5, 0.5 ],
82 "Takemiya Masaki": [ -1.0, 0.0 ]}
84 ###
85 ### Following code consist of expert based knowledge kindly supplied by
86 ### Alexander Dinerstein 3-pro, Motoki Noguchi 7-dan and Vit Brunner 4-dan)
88 ### The vector at each name corresponds with
89 ### (Territory, Orthodox, Aggressiveness, Thickness)
90 questionare_vit_brun = {
91 "Chen Yaoye": (7, 5, 7, 6),
92 "Cho Chikun": (9, 7, 7, 9),
93 "Cho U": (4, 6, 7, 4),
94 "Gu Li": (5, 6, 9, 5),
95 "Ishida Yoshio": (6, 3, 5, 5),
96 "Luo Xihe": (8, 4, 7, 7),
97 "Ma Xiaochun": (5, 7, 7, 7),
98 "O Meien": (3, 9, 6, 5),
99 "Otake Hideo": (4, 3, 6, 5),
100 "Rui Naiwei": (5, 6, 8, 5),
101 "Sakata Eio": (6, 4, 8, 6),
102 "Takemiya Masaki": (1, 4, 7, 2),
103 "Yi Ch'ang-ho 2004-": (7, 6, 4, 4),
104 "Yi Ch'ang-ho 2005+": (7, 6, 6, 4),
105 "Yi Se-tol": (6, 5, 9, 5),
106 "Yoda Norimoto": (4, 4, 7, 3)
108 questionare_motoki_noguchi = {
109 "Cho Chikun": (8, 9, 8, 8 ),
110 "Cho U": (9, 7, 6, 8),
111 "Gu Li": (7, 8, 10, 4 ),
112 "Ishida Yoshio": (9, 6, 2, 6),
113 "Luo Xihe": (6, 8, 9, 7 ),
114 "Ma Xiaochun": (9, 6, 7, 8),
115 "O Meien": (1, 10, 10, 2 ),
116 "Otake Hideo": (4, 3, 5, 3),
117 "Rui Naiwei": (6, 6, 10, 2),
118 "Sakata Eio": (10, 5, 6, 10),
119 "Takemiya Masaki": (2,6, 6, 1),
120 "Yi Ch'ang-ho 2004-": (8, 3, 2, 3),
121 "Yi Se-tol": (5, 10, 10, 8 ),
122 "Yoda Norimoto": (8, 2, 2, 5),
123 "Fujisawa Hideyuki": (4, 8, 7, 4 ),
124 "Go Seigen": (8, 10, 9, 6),
125 "Hane Naoki": (8, 2, 4, 6 ),
126 "Honinbo Dosaku": (2, 10, 8, 5 ),
127 "Honinbo Shusaku": (8, 3, 2, 6),
128 "Honinbo Shuwa": (10, 8, 2, 10),
129 "Kato Masao": (2,3, 9, 4),
130 "Kobayashi Koichi": (8, 3, 3, 6),
131 "Miyazawa Goro": (1, 10, 10, 3),
132 "Takao Shinji": (4, 3, 7, 4 ),
133 "Yamashita Keigo": (2, 8, 10, 4 ),
134 "Yuki Satoshi": (2, 8, 10, 4)
136 questionare_alex_dinner = {
137 "Chen Yaoye": (5, 3, 5, 5),
138 "Cho Chikun": (10, 7, 5, 10),
139 "Cho U": (9, 5, 3, 7),
140 "Gu Li": (5, 7, 8, 3),
141 "Ishida Yoshio": (9, 6, 3, 5),
142 "Luo Xihe": (8, 10, 7, 4),
143 "Ma Xiaochun": (10, 6, 3, 9),
144 "O Meien": (4, 10, 9, 4),
145 "Otake Hideo": (5, 3, 3, 3),
146 "Rui Naiwei": (3, 5, 9, 3),
147 "Sakata Eio": (7, 5, 8, 8),
148 "Takemiya Masaki": (1, 9, 8, 1),
149 "Yi Ch'ang-ho 2004-": (6, 6, 2, 1),
150 "Yi Ch'ang-ho 2005+": (5, 4, 5, 3),
151 "Yi Se-tol": (5, 5, 9, 7),
152 "Yoda Norimoto": (7, 7, 4, 2),
153 "Chen Zude": (3, 8, 6, 5),
154 "Cho Tae-hyeon": (1, 4, 4, 2),
155 "Fujisawa Hideyuki": (3, 10, 7, 4),
156 "Go Seigen": (4, 8, 7, 4),
157 "Hane Naoki": (7, 3, 4, 3),
158 "Jie Li (USA)": (5, 3, 5, 4),
159 "Kato Masao": (3, 6, 10, 4),
160 "Kobayashi Koichi": (10, 2, 2, 5),
161 "Miyazawa Goro": (2, 10, 9, 5),
162 "Nie Weiping": (3, 7, 8, 4),
163 "Shao Zhenzhong": (4, 5, 5, 4),
164 "Suzuki Goro": (4, 7, 5, 5),
165 "Takao Shinji": (6, 4, 4, 5),
166 "Wu Songsheng": (2, 10, 7, 4),
167 "Yamashita Keigo": (2, 10, 9, 2),
168 "Yuki Satoshi": (4, 9, 8, 5),
169 "breakfast": (7, 7, 3, 4),
170 "rapyuta/daien": (4, 7, 6, 5),
171 "MilanMilan": (5, 5, 6, 4),
172 "roln111-": (6, 5, 7, 5),
173 "somerville": (4, 5, 5, 6),
174 "artem92-": (7, 4, 3, 2),
175 "TheCaptain": (3, 8, 7, 6)
177 questionare_list = [ questionare_vit_brun, questionare_motoki_noguchi, questionare_alex_dinner ]
178 questionare_total = questionare_average(questionare_list,silent=True)
180 def questionare_variance():
181 questionare_average(Data.questionare_list)
183 if __name__ == '__main__':
184 import sys
186 for arg in sys.argv:
187 if arg == '-p' or arg == '--all-players':
188 for p in Data.players_all:
189 print p
190 if arg == '-h' or arg == '--help':
191 print sys.argv[0], " [OPTIONS]"
192 print """
194 OPTIONS
195 -p, --all-players
196 prints a list of known players to the STDOUT.
197 -h, --help
198 help!