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()