예제 #1
0
파일: main.py 프로젝트: andoniVT/ESA_Final
    def __testClassifier(self, segmentos, entities, model, fileClass):
        results = []
        for j in segmentos:
            proc = TextCleaner(j)
            procesado = proc.get_processed_comment()

            vector = model.get_comment_tf_idf_vector([procesado])
            supClass = load_data_from_disk(fileClass)
            classifier = SC()
            classifier.set_classifier(supClass)
            result = classifier.classify(vector)

            polaridadSup = result[0][0]
            for i in entities:
                if j.find(i[0]) != -1:
                    value = (i[0], polaridadSup)
                    results.append(value)
        return results