Bayesian Machine Learning Algorithm written in Swift 3.0