aspect_nltk_nb.py 文件源码

def classify(self,features,threshold=0.8):
        business_keywords = ["business","wi","fi","wifi","internet","wireless"]
        clean_keywords = ['clean']

        prob_dist = self.classifier.prob_classify(features)
        most_prob_label = prob_dist.max()

        if prob_dist.prob(most_prob_label) >= threshold:
            return most_prob_label
        elif any( bkeyword in features for bkeyword in business_keywords):
            # features contain keyword for business
            return common.AspectBusiness
        elif any ( ckeyword in features for ckeyword in clean_keywords):
            return common.AspectClean
        else:
            return common.AspectUnknown