Confused on training data

[YanhaoZhang]

Thanks a lot for releasing this code. May I know if you could provide a simple dataset for running the code? Just for making reading the code more easily. My feeling is that the current readme file is a bit confused. Thanks a lot.

[ultmaster]

I think you can read the predict first. It might give you a sense about what's going on in the model.