def state_value(self, state: GameState) -> float:
"""Return the probability of the state being winning from the neural network"""
prob_distribution = self.neural_network.predict_proba([state.vector_representation()])[0]
# The first value in prob_distribution is the probability of the value being -1
# and the second is the probability of the value being 1
if self.is_player1:
return prob_distribution[1]
return prob_distribution[0]
