Add functionality to train different types of ML models
Added ml library and got weights after training. Need to figure out how to init a model with weights.
Added ml library and got weights after training. Need to figure out how to init a model with weights.