Add predict function

[dakinggg]

trafficstars

Hi,

Thank you for the library! I don't know if anyone is maintaining this, but I wanted to use the library and found that I needed a predict function, so I wrote one. If anyone is maintaining this repo, I would love to get feedback on the correctness of my predict function and to integrate it into your model if appropriate. Thanks!