Train on their dataset, test on ours.

Gets 0.58. Ours gets 0.60 using cross validation.
Fix bug with generating kmers_index, which should
be consistent across runs.