if pos_group < num_train:  # train
        idf = 0
    elif pos_group < (num_train + num_val):  # val
        idf = 1
    else:  # test