@ignitionai/backend-tfjs
Version:
TensorFlow.js backend for IgnitionAI - browser-based reinforcement learning framework
16 lines (15 loc) • 350 B
TypeScript
export interface Experience {
state: number[];
action: number;
reward: number;
nextState: number[];
done: boolean;
}
export declare class ReplayBuffer {
private buffer;
private capacity;
constructor(capacity?: number);
add(exp: Experience): void;
sample(batchSize: number): Experience[];
size(): number;
}