https://medium.com/@monocosmo77/working-with-vertical-federated-learning-part4-machine-learning-ff0086d825e5