← Questions

Next-word predictor from bigram counts

0
GoogleCodingLast asked Feb 2026

You get training sentences as lists of words, for example [["a", "b", "c"], ["a", "s", "d"]].

  1. Count, for every word, which words follow it and how often.
  2. predict(word) returns the word that most often follows it.
  3. Change predict to sample the next word at random, weighted by those counts.
  4. The vocabulary is huge. How do you keep memory and prediction time under control?
Write and test your solution herePython in your browser, the question's tests level by level, every submission kept for you.Sign in to start coding