def train_test_split(tokenized):
    """Returns a train, test tuple of dataframes"""
    train = tokenized.query("category != 'title'")
    test = tokenized.query("category == 'title'")
    return train, test