Added function to create train/test split data
LDA wasn't actually using custom hyper-params
Logged LDA hyper-params and fixed bug with generating train/test data