@ignitionai/backend-tfjs/dist/tools/trainer.d.ts

TensorFlow.js backend for IgnitionAI - browser-based reinforcement learning framework

import { DQNConfig } from "../types";
interface TrainAgentOptions {
    config: DQNConfig;
    maxSteps?: number;
    checkpointEvery?: number;
    repoId?: string;
    token?: string;
    onStep?: (step: number, reward: number, action: number, state: number[]) => void;
    getEnvStep: () => {
        state: number[];
        correctAction: number;
        nextState: number[];
    };
}
export declare function trainAgent(options: TrainAgentOptions): Promise<void>;
export {};