Remove NA on no friends
For train run, Fix the test data and randomise sample data to test for variance and bias