96 lines
3.8 KiB
Python
96 lines
3.8 KiB
Python
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 app == 0.0:
|
|
continue
|
|
if vote == 1:
|
|
total = total * app
|
|
else:
|
|
total = total * (1.0 - app)
|
|
#print total
|
|
return total
|
|
|
|
|
|
|
|
|
|
|