class SentimentClassifierTestCase(unittest.TestCase): def setUp(self): base_dir = os.path.dirname(__file__) self.classifier = SentimentClassifier(SegTagger(), os.path.join(base_dir, "./model/sentiment.svm.model"), os.path.join(base_dir, "./model/sentiment_feature_index.json")) def tearDown(self): pass def testSentimentClassifier(self): res = self.classifier.classifier(u"标准间太差 房间还不如3星的 而且设施非常陈旧.建议酒店把老的标准间从新改善.") self.assertEqual(res, 0) res = self.classifier.classifier(u"距离川沙公路较近,但是公交指示不对,如果是\"蔡陆线\"的话,会非常麻烦.建议用别的路线.房间较为简单.") self.assertEqual(res, 1) res = self.classifier.classifier(u"预订标准间:拐角房间(为什么携程订的都是拐角间),房间很小,隔音超差,房间非常冷,空调几乎不好用,卫生间更冷,几乎没法淋浴!三个字"不满意"!") self.assertEqual(res, 0) res = self.classifier.classifier(u"商务大床房,房间很大,床有2M宽,整体感觉经济实惠不错!") self.assertEqual(res, 1)
class SentimentClassifierTestCase(unittest.TestCase): def setUp(self): base_dir = os.path.dirname(__file__) self.classifier = SentimentClassifier( SegTagger(), os.path.join(base_dir, "./model/sentiment.svm.model"), os.path.join(base_dir, "./model/sentiment_feature_index.json")) def tearDown(self): pass def testSentimentClassifier(self): res = self.classifier.classifier( u"标准间太差 房间还不如3星的 而且设施非常陈旧.建议酒店把老的标准间从新改善.") self.assertEqual(res, 0) res = self.classifier.classifier( u"距离川沙公路较近,但是公交指示不对,如果是\"蔡陆线\"的话,会非常麻烦.建议用别的路线.房间较为简单.") self.assertEqual(res, 1) res = self.classifier.classifier( u"预订标准间:拐角房间(为什么携程订的都是拐角间),房间很小,隔音超差,房间非常冷,空调几乎不好用,卫生间更冷,几乎没法淋浴!三个字"不满意"!" ) self.assertEqual(res, 0) res = self.classifier.classifier(u"商务大床房,房间很大,床有2M宽,整体感觉经济实惠不错!") self.assertEqual(res, 1)
def setUp(self): base_dir = os.path.dirname(__file__) self.classifier = SentimentClassifier(SegTagger(), os.path.join(base_dir, "./model/sentiment.svm.model"), os.path.join(base_dir, "./model/sentiment_feature_index.json"))
def sentiment(txt): global sentiment_classifier if sentiment_classifier is None: sentiment_classifier = SentimentClassifier(seg_tagger) return str(int(sentiment_classifier.classifier(txt)))
def setUp(self): base_dir = os.path.dirname(__file__) self.classifier = SentimentClassifier( SegTagger(), os.path.join(base_dir, "./model/sentiment.svm.model"), os.path.join(base_dir, "./model/sentiment_feature_index.json"))