Comment by jonfw

> * Way way way more code in the training set.

Why not convert the training code to AST?