ml-q-learning
Version:
Library implementing the q-learning algorithm and several exploration algorithms.
15 lines (14 loc) • 476 B
TypeScript
import { IMemoryAdapter } from "./memory";
import { IStep, IState } from "./q-learning-agent";
export interface IQLearningAgent<TAction = any> {
replayMemory: [string, number, number][];
episode: number;
trained: boolean;
actions: TAction[];
memory: IMemoryAdapter;
learningRate: number;
discountFactor: number;
play(state: IState): Promise<IStep<TAction>>;
reward(step: IStep<TAction>, reward: number): void;
learn(): Promise<void>;
}