Predicting the Formation of Induction Heads