nesta-gitu / simple-q Goto Github PK

Q-Learning

A low learning curve implementation of Q-learning.

.

Q-learning is a reinforcement learning algorithm that allows an agent to learn an optimal policy for sequential decision-making problems. This implementation provides an easy-to-use and beginner-friendly approach to Q-learning, making it accessible to users with varying levels of experience. This implementation is build to work with "GYM" environments from the Gymnasium Project.

• Provides flexibility to work with either integer or custom objects for states and actions.
• Offers customizable learning rate and exploration-exploitation trade-off parameters.
• Implements epsilon-greedy strategy for action selection.
• Supports polynomial and power decay functions for learning rate and exploration parameter respecitively. The learning rate decay is based on the research in Even-Dar et al. (2003). When lambda is in (0,1) this implementation of Q-learning satisfies the conditions for probability one convergence in Watkins and Dayan (1992).

Install the Simple_Q package from PyPI:

pip install Simple_Q

from Simple_Q import Qlearning

## Usage

```python
from Simple_Q import Qlearning

# Create the Q-learning agent
agent = Qlearning(states=10, actions = ['forward', 'left', 'right' 'stop'])

# Use the Q-learning agent in your reinforcement learning loop
state = env.reset()
N = 100
for n in range(N):
    action = agent.get_action(state)
    next_state, reward, done, _ = env.step(action)
    agent.update_q_table(state, action, reward, next_state)
    state = next_state
    agent.update_epsilon()

See also the Savings Problem example in the GitHub repository.

The Q-Learning class provides the following parameters:

The Q-Learning class provides the following attributes:

Contributions are welcome! If you find any issues or have suggestions for improvement, please feel free to open an issue or submit a pull request.

This project is licensed under the MIT License.

This code is originally based upon the Q-Learning implementation from the book: "Advanced Deep Learning with TensorFlow 2 and Keras" by Rowel Atienza. However many changes have been made to the original code, including the addition of the polynomial learning rate decay as well as the ability to work with custom objects for states and actions.

Even-Dar, E., Y. Mansour, and P. Bartlett (2003). Learning rates for q-learning. Journal of machine learning Research 5(1).

Watkins, C. J. and P. Dayan (1992, May). Technical note: Q-learning. Machine Learning 8(3), 279–292