wip on bayes
datasets for training