added in case for train_test_negative_set in utilities.py