Make learning activity more robust and make its tests more robust

[Nitwix]

From the bug column in the project board

Learning on datasets with more than 3 categories can put the user in an impossible position.