def METHOD_NAME(self, batch_data):
    input_feat = [
        batch_data["user_index_batch"],
        batch_data["clicked_title_batch"],
        batch_data["candidate_title_batch"],
    ]
    input_label = batch_data["labels"]
    return input_feat, input_label