mu_zero/example3 library

Functions

argMax(List<double> list) → int

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

main() → void

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