Implementing further tests for the supervised learning stage.
Added a visualisation for the test set.