Problem solved for ValueError in FeatureHasher and naive bayes.

nineil · Jun 30, 2014 · 760e7ab · 760e7ab
1 parent 43b27a6
commit 760e7ab
Showing 1 changed file with 0 additions and 9 deletions.
diff --git a/tests/test_skll.py b/tests/test_skll.py
@@ -544,12 +544,6 @@ def test_summary_feature_hasher():
         outstr = f.read()
         logistic_result_score = float(SCORE_OUTPUT_RE.search(outstr).groups()[0])
 
-    with open(os.path.join(_my_dir, 'output',
-                           'test_summary_feature_hasher_test_summary_MultinomialNB.results')) \
-            as f:
-        outstr = f.read()
-        naivebayes_result_score = float(SCORE_OUTPUT_RE.search(outstr).groups()[0])
-
     with open(os.path.join(_my_dir, 'output',
                            'test_summary_feature_hasher_test_summary_SVC.results')) \
             as f:
@@ -572,13 +566,10 @@ def test_summary_feature_hasher():
 
             if row['learner_name'] == 'LogisticRegression':
                 logistic_summary_score = float(row['score'])
-            elif row['learner_name'] == 'MultinomialNB':
-                naivebayes_summary_score = float(row['score'])
             elif row['learner_name'] == 'SVC':
                 svm_summary_score = float(row['score'])
 
     for result_score, summary_score, learner_name in [(logistic_result_score, logistic_summary_score, 'LogisticRegression'),
-                                                      (naivebayes_result_score, naivebayes_summary_score, 'MultinomialNB'),
                                                       (svm_result_score, svm_summary_score, 'SVC')]:
         yield check_summary_score, result_score, summary_score, learner_name