Add a real model
This commit is contained in:
parent
12316a9238
commit
8b7cab4c59
7 changed files with 432 additions and 10 deletions
93
models/data_model.py
Normal file
93
models/data_model.py
Normal file
|
|
@ -0,0 +1,93 @@
|
|||
from default_model import DefaultModel
|
||||
from probs_from_game import TrainTable, approval
|
||||
|
||||
translate = {
|
||||
"Merlin": "merlin",
|
||||
"Morgana": "morgana",
|
||||
"Oberon": "oberon",
|
||||
"Mordred": "mordred",
|
||||
"Percival": "percival",
|
||||
"GLance": "good",
|
||||
"G": "good",
|
||||
"ELance": "assassin"
|
||||
}
|
||||
|
||||
class DataModel(DefaultModel):
|
||||
def __init__(self, game):
|
||||
super(DataModel, self).__init__(game)
|
||||
self.table = TrainTable("models/dataset.json")
|
||||
|
||||
#def mission(self, team, fails, must_fail, rnd):
|
||||
#pass
|
||||
|
||||
def votes(self, team, votes, fail_req, rnd, voten, proposer):
|
||||
total = 1.0
|
||||
n_actually_good_people = sum(
|
||||
[int(self.is_good(x, rnd)) for x in team])
|
||||
n_spies = len(team) - n_actually_good_people
|
||||
|
||||
prole = self.player_role(proposer)
|
||||
for player, vote in enumerate(votes):
|
||||
role = self.player_role(player)
|
||||
if role == "Merlin":
|
||||
m_spies = n_spies
|
||||
roles = [self.player_role(x) for x in team]
|
||||
if "Mordred" in roles:
|
||||
m_spies -= 1
|
||||
data = self.table.MerlinVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
known_evil=m_spies,
|
||||
size=len(votes))
|
||||
elif role == "Percival":
|
||||
roles = [self.player_role(x) for x in team]
|
||||
tc = ""
|
||||
if "merlin" in roles:
|
||||
tc += "merlin"
|
||||
if "morgana" in roles:
|
||||
tc += "morgana"
|
||||
if tc == "":
|
||||
tc = "neither"
|
||||
data = self.table.PercivalVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
team_contains=tc,
|
||||
size=len(votes))
|
||||
elif role == "G" or role == "GLance":
|
||||
data = self.table.GoodVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
size=len(votes))
|
||||
elif role == "Mordred":
|
||||
data = self.table.MordredVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
size=len(votes))
|
||||
elif role == "Morgana":
|
||||
data = self.table.MorganaVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
size=len(votes))
|
||||
elif role == "Oberon":
|
||||
data = self.table.OberonVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
size=len(votes))
|
||||
elif role == "ELance" or "E":
|
||||
data = self.table.EvilVote(rnd + 1, voten + 1,
|
||||
proposer_role=translate[prole],
|
||||
evil_count=n_spies,
|
||||
size=len(votes))
|
||||
else:
|
||||
continue
|
||||
app = approval(data)
|
||||
if vote == 1:
|
||||
total = total * app
|
||||
else:
|
||||
total = total * (1.0 - app)
|
||||
return total
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
BIN
models/dataset.json.gz
Normal file
BIN
models/dataset.json.gz
Normal file
Binary file not shown.
|
|
@ -51,7 +51,7 @@ class BaseModel(object):
|
|||
def mission(self, team, fails, must_fail, rnd):
|
||||
return True
|
||||
|
||||
def votes(self, team, votes, fail_req, rnd):
|
||||
def votes(self, team, votes, fail_req, rnd, voten, proposer):
|
||||
return True
|
||||
|
||||
|
||||
|
|
@ -130,7 +130,7 @@ class DefaultModel(BaseModel):
|
|||
return 1.0
|
||||
return 1.0
|
||||
|
||||
def votes(self, team, votes, fail_req, rnd):
|
||||
def votes(self, team, votes, fail_req, rnd, voten, proposer):
|
||||
for player, vote in enumerate(votes):
|
||||
role = self.player_role(player)
|
||||
if role in special_votes:
|
||||
|
|
|
|||
204
models/probs_from_game.py
Normal file
204
models/probs_from_game.py
Normal file
|
|
@ -0,0 +1,204 @@
|
|||
from collections import defaultdict as dd
|
||||
import json
|
||||
|
||||
roles = [u'oberon', u'merlin', u'mordred', u'morgana', u'percival', u'assassin']
|
||||
|
||||
ANY_GOOD = [u'merlin', u'percival', u'good']
|
||||
ANY_EVIL = [u'oberon', u'mordred', u'morgana', u'assassin', u'evil']
|
||||
|
||||
|
||||
def isSpy(x, game):
|
||||
return game["players"][x]["spy"]
|
||||
|
||||
def approval(x):
|
||||
if x["Approve"] + x["Reject"] == 0:
|
||||
return 0.0
|
||||
return (x["Approve"] * 1.0) / (x["Approve"] + x["Reject"])
|
||||
|
||||
def getRole(x, game):
|
||||
if str(x) not in game["role_by_seat"]:
|
||||
if isSpy(x, game):
|
||||
return "evil"
|
||||
return "good"
|
||||
return game["role_by_seat"][str(x)]
|
||||
|
||||
def sumKeys(a, b):
|
||||
out = dd(int)
|
||||
for k, v in a.items():
|
||||
out[k] += v
|
||||
for k, v in b.items():
|
||||
out[k] += v
|
||||
return out
|
||||
|
||||
def sum_helper(table, constraint_list):
|
||||
if len(constraint_list) == 0:
|
||||
return table
|
||||
c = constraint_list[0]
|
||||
out = dd(int)
|
||||
if isinstance(c, type([])):
|
||||
for v in c:
|
||||
out = sumKeys(out, sum_helper(table[v], constraint_list[1:]))
|
||||
elif c is None:
|
||||
for k in table:
|
||||
out = sumKeys(out, sum_helper(table[k], constraint_list[1:]))
|
||||
else:
|
||||
return sumKeys(out, sum_helper(table[c], constraint_list[1:]))
|
||||
return out
|
||||
|
||||
class TrainTable(object):
|
||||
"""Docstring for TrainTable. """
|
||||
|
||||
# good: round, vote, proposer_role, evils on team
|
||||
def __init__(self, path):
|
||||
self.good_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int))))))
|
||||
self.percy_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int)))))))
|
||||
self.merlin_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int)))))))
|
||||
self.evil_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int))))))
|
||||
self.morgana_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int))))))
|
||||
self.assassin_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int))))))
|
||||
self.oberon_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int))))))
|
||||
self.mordred_vote = dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(lambda: dd(int))))))
|
||||
self._path = path
|
||||
self._build_prob_table()
|
||||
|
||||
def GoodVote(self, round=None, vote=None, proposer_role=None, evil_count=None, size=None):
|
||||
return sum_helper(self.good_vote, [size, round, vote, proposer_role, evil_count])
|
||||
|
||||
def PercivalVote(self, round=None, vote=None, proposer_role=None, evil_count=None, team_contains=None, size=None):
|
||||
return sum_helper(self.percy_vote, [size, round, vote, proposer_role, evil_count, team_contains])
|
||||
|
||||
def EvilVote(self, round=None, vote=None, proposer_role=None, evil_count=None, size=None):
|
||||
return sum_helper(self.evil_vote, [size, round, vote, proposer_role, evil_count])
|
||||
|
||||
def MordredVote(self, round=None, vote=None, proposer_role=None, evil_count=None, size=None):
|
||||
return sum_helper(self.mordred_vote, [size, round, vote, proposer_role, evil_count])
|
||||
|
||||
def MorganaVote(self, round=None, vote=None, proposer_role=None, evil_count=None, size=None):
|
||||
return sum_helper(self.morgana_vote, [size, round, vote, proposer_role, evil_count])
|
||||
|
||||
def AssassinVote(self, round=None, vote=None, proposer_role=None, evil_count=None, size=None):
|
||||
return sum_helper(self.asassin_vote, [size, round, vote, proposer_role, evil_count])
|
||||
|
||||
def OberonVote(self, round=None, vote=None, proposer_role=None, evil_count=None, size=None):
|
||||
return sum_helper(self.oberon_vote, [size, round, vote, proposer_role, evil_count])
|
||||
|
||||
def MerlinVote(self, round=None, vote=None, proposer_role=None, evil_count=None, known_evil=None, size=None):
|
||||
return sum_helper(self.merlin_vote, [size, round, vote, proposer_role, evil_count, known_evil])
|
||||
|
||||
|
||||
def _build_prob_table(self):
|
||||
f = open(self._path)
|
||||
self._data = json.load(f)
|
||||
f.close()
|
||||
self._build_good_vote(self.good_vote, "good")
|
||||
self._build_percy_vote(self.percy_vote)
|
||||
self._build_merlin_vote(self.merlin_vote)
|
||||
self._build_evil_vote(self.evil_vote, "evil")
|
||||
self._build_evil_vote(self.mordred_vote, "mordred")
|
||||
self._build_evil_vote(self.morgana_vote, "morgana")
|
||||
self._build_evil_vote(self.assassin_vote, "assassin")
|
||||
self._build_evil_vote(self.oberon_vote, "oberon")
|
||||
|
||||
def _build_good_vote(self, table, exprole):
|
||||
for game in self._data:
|
||||
size = len(game["players"])
|
||||
for player in game["players"]:
|
||||
idx = player["seat"]
|
||||
if player["spy"]:
|
||||
continue
|
||||
if getRole(idx, game) != exprole:
|
||||
continue
|
||||
for round_n, round in enumerate(game["log"]):
|
||||
for vote_n in range(1, 6):
|
||||
if str(vote_n) not in round:
|
||||
break
|
||||
vote = round[str(vote_n)]
|
||||
prole = getRole(vote["proposer"], game)
|
||||
if vote["proposer"] == idx:
|
||||
prole = "self"
|
||||
evils = sum([int(isSpy(x, game)) for x in vote["team"]])
|
||||
action = vote["votes"][idx]
|
||||
table[size][round_n + 1][vote_n][prole][evils][action] += 1
|
||||
|
||||
def _build_evil_vote(self, table, exprole):
|
||||
for game in self._data:
|
||||
size = len(game["players"])
|
||||
for player in game["players"]:
|
||||
idx = player["seat"]
|
||||
if not player["spy"]:
|
||||
continue
|
||||
if getRole(idx, game) != exprole:
|
||||
continue
|
||||
for round_n, round in enumerate(game["log"]):
|
||||
for vote_n in range(1, 6):
|
||||
if str(vote_n) not in round:
|
||||
break
|
||||
vote = round[str(vote_n)]
|
||||
prole = getRole(vote["proposer"], game)
|
||||
if vote["proposer"] == idx:
|
||||
prole = "self"
|
||||
evils = sum([int(isSpy(x, game)) for x in vote["team"]])
|
||||
action = vote["votes"][idx]
|
||||
table[size][round_n + 1][vote_n][prole][evils][action] += 1
|
||||
|
||||
def _build_percy_vote(self, table):
|
||||
for game in self._data:
|
||||
size = len(game["players"])
|
||||
for player in game["players"]:
|
||||
idx = player["seat"]
|
||||
if player["spy"]:
|
||||
continue
|
||||
if getRole(idx, game) != "percival":
|
||||
continue
|
||||
for round_n, round in enumerate(game["log"]):
|
||||
for vote_n in range(1, 6):
|
||||
if str(vote_n) not in round:
|
||||
break
|
||||
vote = round[str(vote_n)]
|
||||
prole = getRole(vote["proposer"], game)
|
||||
evils = sum([int(isSpy(x, game)) for x in vote["team"]])
|
||||
action = vote["votes"][idx]
|
||||
tc = ""
|
||||
rolesOnTeam = [getRole(x, game) for x in vote["team"]]
|
||||
if "merlin" in rolesOnTeam:
|
||||
tc += "merlin"
|
||||
if "morgana" in rolesOnTeam:
|
||||
tc += "morgana"
|
||||
if tc == "":
|
||||
tc = "neither"
|
||||
table[size][round_n + 1][vote_n][prole][evils][tc][action] += 1
|
||||
|
||||
def _build_merlin_vote(self, table):
|
||||
for game in self._data:
|
||||
size = len(game["players"])
|
||||
for player in game["players"]:
|
||||
idx = player["seat"]
|
||||
if player["spy"]:
|
||||
continue
|
||||
if getRole(idx, game) != "merlin":
|
||||
continue
|
||||
for round_n, round in enumerate(game["log"]):
|
||||
for vote_n in range(1, 6):
|
||||
if str(vote_n) not in round:
|
||||
break
|
||||
vote = round[str(vote_n)]
|
||||
prole = getRole(vote["proposer"], game)
|
||||
evils = sum([int(isSpy(x, game)) for x in vote["team"]])
|
||||
action = vote["votes"][idx]
|
||||
known_evils = sum([int(isSpy(x, game) and (getRole(x, game) != "mordred")) for x in vote["team"]])
|
||||
table[size][round_n + 1][vote_n][prole][evils][known_evils][action] += 1
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
def main():
|
||||
table = TrainTable("dataset.json")
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue