mu_zero/example library

Functions

argMax(List<double> list) → int

generateMuZeroGreedy(MuZeroGreedyAgent agent, List<int> prompt, int maxLength, Map<int, String> itos) → void

main() → void

trainMuZero(MuZeroGreedyAgent agent, List<int> targetSequence, int epochs) → void