@mindfulai/rl-engine

A lightweight Reinforcement Learning engine for NPM usage, built with Q-learning logic.

Features

• Q-table initialization
• ε-greedy action selection
• Q-learning update
• Easily configurable

Installation

npm install @mindfulai/rl-engine

Usage

import { initQTable, RLAgent } from "@mindfulai/rl-engine";

const states = ["happy", "sad", "neutral"];
const actions = ["suggest_music", "recommend_meditation", "show_quote"];
const qTable = initQTable(states, actions);

const agent = new RLAgent(qTable);

const state = "sad";
const action = agent.chooseAction(state);
agent.update(state, action, 1, "neutral");

console.log(agent.getQTable());