mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-22 16:48:57 +00:00
moved to community folder
This commit is contained in:
parent
a17dbdfedc
commit
0f61c9dbde
24 changed files with 118 additions and 9 deletions
|
|
@ -0,0 +1,18 @@
|
|||
import { getAggressiveMove } from "../agents/attacker_agent";
|
||||
import { ChessEnv } from "../env/chess_env";
|
||||
|
||||
export async function runTrainingLoop(episodes = 1) {
|
||||
const env = new ChessEnv();
|
||||
for (let ep = 0; ep < episodes; ep++) {
|
||||
env.reset();
|
||||
let done = false;
|
||||
while (!done) {
|
||||
const move = getAggressiveMove(env.getFEN());
|
||||
if (!move) break;
|
||||
const { done: isDone } = env.step(move);
|
||||
done = isDone;
|
||||
}
|
||||
// Log or store game
|
||||
console.log(`Episode ${ep + 1} finished. FEN: ${env.getFEN()}`);
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue