aditya-epsgreedy-20251112 0.0.1

A tiny, well-tested epsilon-greedy action selector for Reinforcement Learning with action masking. Created by Aditya Singh for B.Tech AI (RL Practicum, NMIMS MPSTME).

Epsilon-Greedy

A minimal, tested epsilon-greedy action selector for Reinforcement Learning. Includes action masking and tie-aware probability allocation.

Usage

from epsgreedy import epsilon_greedy

action_values = [1.2, 0.5, 1.2, -0.3]   # Q(s, ·)
# Mask out invalid actions (e.g., action 1 is invalid here)
invalid = [False, True, False, False]

a, p = epsilon_greedy(action_values, epsilon=0.1, invalid=invalid)
print(a, p)  # chosen action index, probabilities over actions

Install

pip install aditya-epsgreedy-20251112

Testing

pip install pytest
pytest -q

License

MIT