machine learning final project - text prediction