def main():
#--------------------------------------------------------------------------
# 2-state Markov chain
#--------------------------------------------------------------------------
P = np.array([[0.8, 0.2], [0.1, 0.9]]) # Transition matrix
mc = MarkovChain(P, ['1', '2'])
mc.draw("../img/markov-chain-two-states.png")
#--------------------------------------------------------------------------
# 3-state Markov chain
#--------------------------------------------------------------------------
P = np.array([
[0.8, 0.1, 0.1],
[0.1, 0.7, 0.2],
[0.1, 0.7, 0.2],
])
mc = MarkovChain(P, ['A', 'B', 'C'])
mc.draw("../img/markov-chain-three-states.png")
#--------------------------------------------------------------------------
# 4-state Markov chain
#--------------------------------------------------------------------------
P = np.array([[0.8, 0.1, 0.1, 0.0], [0.1, 0.7, 0.0, 0.2],
[0.1, 0.0, 0.7, 0.2], [0.1, 0.0, 0.7, 0.2]])
mc = MarkovChain(P, ['1', '2', '3', '4'])
mc.draw("../img/markov-chain-four-states.png")
def plot_graph(self, mk_prop, Labels):
P = np.array([mk_prop]) # Transition matrix
mc = MarkovChain(P, Labels)
mc.draw()
