def METHOD_NAME(self, doc_ids: List[str]):
    train_docs = doc_ids[:self.r] + doc_ids[-self.n:]
    train_labels = [1] * self.r + [0] * self.n
    train_vecs = self.vectorizer.get_vectors(train_docs)
    test_vecs = self.vectorizer.get_vectors(doc_ids)
    return train_vecs, train_labels, test_vecs