keyterm_classification.py 文件源码

def __init__(self, classes=None, classesFile=None,
                 classesClusterPath=None,
                 modelPath="dataset/frWac_non_lem_no_postag_no_phrase_200_cbow_cut100.bin", modelBinary=True):

        if not(classes) and not(classesFile):
            print "ERROR MUST LOAD CLASS FILE"
            return 202

        if not(classes):
            classes = self.load_adv_keyterms_from_file(classesFile)

        #load cluster
        if classesClusterPath:
            self.classesClusters = load_cluster_dataset(classesClusterPath)
        else:
            #process cluster from classes
            #TODO
            self.classesClusters = None

        self.model = gensim.models.Word2Vec.load_word2vec_format(modelPath, binary=modelBinary)
        self._preProcessClasses(classes)