4571bbf0507e935510b1a89dc6510ee574879fda,qanta/guesser/mem_nn.py,MemNNGuesser,train,#MemNNGuesser#Any#,207

Before Change

    def train(self, training_data: TrainingData) -> None:
        log.info("Collecting Wikipedia data...")
        classes = {format_guess(g) for g in training_data[1]}
        class_sentences = fetch_wikipedia_sentences(classes, self.n_wiki_sentences)

        log.info("Preprocessing training data...")
        x_train, y_train, _, x_test, y_test, _, vocab, class_to_i, i_to_class = preprocess_dataset(training_data)

After Change

    def train(self, training_data: TrainingData) -> None:
        log.info("Collecting Wikipedia data...")
        for i in range(len(training_data[1])):
            training_data[1][i] = format_guess(training_data[1][i])
        classes = set(training_data[1])
        wiki_sentences = build_wikipedia_sentences(classes, self.n_wiki_sentences)

Instances