def prepareModels(self, xml_file, type): comentarios = self.procesar(xml_file, type) train = [] for i in comentarios: train.append(i[0]) model = VM(train) vectorModelData = model.prepare_models() modelVectorizer = vectorModelData[0] modelVectorizerTFIDF = vectorModelData[1] modelTFIDF = vectorModelData[2] if type == 1: write_data_to_disk(simpleVectorizer, modelVectorizer) write_data_to_disk(tfidfVectorizer, modelVectorizerTFIDF) write_data_to_disk(tfidfModel, modelTFIDF) else: write_data_to_disk(simpleVectorizerp, modelVectorizer) write_data_to_disk(tfidfVectorizerp, modelVectorizerTFIDF) write_data_to_disk(tfidfModelp, modelTFIDF)