class TrueSkillFactorGraph(FactorGraph):
def __init__(self, gameInfo, teams, teamRanks):
self._priorLayer = PlayerPriorValuesToSkillsLayer(self, teams)
self._gameInfo = gameInfo
newFactory = VariableFactory(self.fromPrecisionMean)
self.setVariableFactory(newFactory)
self._layers = [
self._priorLayer,
PlayerSkillsToPerformancesLayer(self),
PlayerPerformancesToTeamPerformancesLayer(self),
IteratedTeamDifferencesInnerLayer(
self,
TeamPerformancesToTeamPerformanceDifferencesLayer(self),
TeamDifferencesComparisonLayer(self, teamRanks),
),
]
def fromPrecisionMean(self):
return GaussianDistribution.fromPrecisionMean(0, 0)
def getGameInfo(self):
return self._gameInfo
def buildGraph(self):
lastOutput = None
layers = self._layers
for currentLayer in layers:
if lastOutput != None:
currentLayer.setInputVariablesGroups(lastOutput)
currentLayer.buildLayer()
lastOutput = currentLayer.getOutputVariablesGroups()
def runSchedule(self):
fullSchedule = self.createFullSchedule()
fullScheduleDelta = fullSchedule.visit()
def getProbabilityOfRanking(self):
factorList = FactorList()
layers = self._layers
for currentLayer in layers:
localFactors = currentLayer.getLocalFactors()
for currentFactor in localFactors:
localCurrentFactor = currentFactor
factorList.addFactor(localCurrentFactor)
logZ = factorList.getLogNormalization()
return exp(logZ)
def createFullSchedule(self):
fullSchedule = []
layers = self._layers
for currentLayer in layers:
currentPriorSchedule = currentLayer.createPriorSchedule()
if currentPriorSchedule != None:
fullSchedule.append(currentPriorSchedule)
allLayersReverse = ReversableList(self._layers)
for currentLayer in allLayersReverse.reverse():
currentPosteriorSchedule = currentLayer.createPosteriorSchedule()
if currentPosteriorSchedule != None:
fullSchedule.append(currentPosteriorSchedule)
return ScheduleSequence("Full schedule", fullSchedule)
def getUpdatedRatings(self):
result = RatingContainer()
priorLayerOutputVariablesGroups = self._priorLayer.getOutputVariablesGroups()
for currentTeam in priorLayerOutputVariablesGroups:
for currentPlayer in currentTeam:
localCurrentPlayer = currentPlayer.getKey()
newRating = Rating(currentPlayer.getValue().getMean(), currentPlayer.getValue().getStandardDeviation())
result.setRating(localCurrentPlayer, newRating)
return result
class TrueSkillFactorGraph(FactorGraph):
def __init__(self, gameInfo, teams, teamRanks):
self._priorLayer = PlayerPriorValuesToSkillsLayer(self, teams)
self._gameInfo = gameInfo
newFactory = VariableFactory(self.fromPrecisionMean)
self.setVariableFactory(newFactory)
self._layers = [
self._priorLayer,
PlayerSkillsToPerformancesLayer(self),
PlayerPerformancesToTeamPerformancesLayer(self),
IteratedTeamDifferencesInnerLayer(
self, TeamPerformancesToTeamPerformanceDifferencesLayer(self),
TeamDifferencesComparisonLayer(self, teamRanks))
]
def fromPrecisionMean(self):
return GaussianDistribution.fromPrecisionMean(0, 0)
def getGameInfo(self):
return self._gameInfo
def buildGraph(self):
lastOutput = None
layers = self._layers
for currentLayer in layers:
if (lastOutput != None):
currentLayer.setInputVariablesGroups(lastOutput)
currentLayer.buildLayer()
lastOutput = currentLayer.getOutputVariablesGroups()
def runSchedule(self):
fullSchedule = self.createFullSchedule()
fullScheduleDelta = fullSchedule.visit()
def getProbabilityOfRanking(self):
factorList = FactorList()
layers = self._layers
for currentLayer in layers:
localFactors = currentLayer.getLocalFactors()
for currentFactor in localFactors:
localCurrentFactor = currentFactor
factorList.addFactor(localCurrentFactor)
logZ = factorList.getLogNormalization()
return exp(logZ)
def createFullSchedule(self):
fullSchedule = []
layers = self._layers
for currentLayer in layers:
currentPriorSchedule = currentLayer.createPriorSchedule()
if currentPriorSchedule != None:
fullSchedule.append(currentPriorSchedule)
allLayersReverse = ReversableList(self._layers)
for currentLayer in allLayersReverse.reverse():
currentPosteriorSchedule = currentLayer.createPosteriorSchedule()
if (currentPosteriorSchedule != None):
fullSchedule.append(currentPosteriorSchedule)
return ScheduleSequence("Full schedule", fullSchedule)
def getUpdatedRatings(self):
result = RatingContainer()
priorLayerOutputVariablesGroups = self._priorLayer.getOutputVariablesGroups(
)
for currentTeam in priorLayerOutputVariablesGroups:
for currentPlayer in currentTeam:
localCurrentPlayer = currentPlayer.getKey()
newRating = Rating(
currentPlayer.getValue().getMean(),
currentPlayer.getValue().getStandardDeviation())
result.setRating(localCurrentPlayer, newRating)
return result
