Skip to Content
Hands-On Machine Learning for Algorithmic Trading
book

Hands-On Machine Learning for Algorithmic Trading

by Stefan Jansen
December 2018
Beginner to intermediate
684 pages
21h 9m
English
Packt Publishing
Content preview from Hands-On Machine Learning for Algorithmic Trading

Solving MDPs using pymdptoolbox

We can also solve MDPs using the pymdptoolbox Python library, which includes a few more algorithms, including Q-learning.

To run ValueIteration, just instantiate the corresponding object with the desired configuration options and the rewards and transition matrices before calling the .run() method:

vi = mdp.ValueIteration(transitions=transitions,                        reward=rewards,                        discount=gamma,                        epsilon=epsilon)vi.run()

The value function estimate matches the result in the previous section:

np.allclose(V.reshape(grid_size), np.asarray(vi.V).reshape(grid_size))

The PolicyIteration function works similarly:

pi = mdp.PolicyIteration(transitions=transitions,                         reward=rewards,                         discount=gamma,                         max_iter=1000)pi.run()

It also yields the ...

Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Start your free trial

You might also like

Machine Learning for Algorithmic Trading - Second Edition

Machine Learning for Algorithmic Trading - Second Edition

Stefan Jansen

Publisher Resources

ISBN: 9781789346411Supplemental Content