Adjust the train_test_split parts in pipelines

[stavrostheocharis]

We use "train_test_split" in "src/analytics/models/pipelines" at lines 40 & 98. In case that after the splitting only one class remains at the train set, an error will follow based on the next calculations (AUC ROC, etc.).

Need to adjust the train_test_split parts and possibly use the "stratify" argument as described inside the docs: https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.train_test_split.html

[aditkay95]

I am willing to work on this as my first issue @stavrostheocharis @momegas

[momegas]

Feel free to open a PR. I assign this to you as well.

[momegas]

Hey @aditkay95 please tell me when to move to in progress when you start working on it

[Mikehade]

I would like to work on this issue @momegas and @stavrostheocharis

[momegas]

Im assining this to you