def load_data(loc='./'):
    """
    Load the TREC question-type dataset
    """
    train, test = [], []
    with open(loc + 'train_5500.label', 'rb') as f:
        for line in f:
            train.append(line.strip())
    with open(loc + 'TREC_10.label', 'rb') as f:
        for line in f:
            test.append(line.strip())
    return train, test