main function

void main()

Implementation

void main() {
  print("--- Stable Tensor-Engine MuZero-GPT: Training & Generation ---");

  final Map<String, int> stoi = {
    "hello": 0,
    "world": 1,
    ".": 2,
    "<start>": 3,
    "<pad>": 4,
  };
  final Map<int, String> itos = stoi.map((k, v) => MapEntry(v, k));

  final model = TransformerDecoder(
    vocabSize: 5,
    embedSize: 32,
    encoderEmbedSize: 32,
    blockSize: 16,
    numLayers: 2,
    numHeads: 4,
  );
  final agent = MuZeroGreedyAgent(model, 32);

  final List<int> trainingData = [
    stoi["<start>"]!,
    stoi["hello"]!,
    stoi["world"]!,
    stoi["."]!,
  ];

  // --- TRAIN FOR MORE THAN 0 EPOCHS ---
  trainMuZero(agent, trainingData, 100); // 500 epochs to actually learn

  print("\n--- Generation After Training ---");
  generateMuZeroGreedy(agent, [stoi["<start>"]!], 4, itos);
}