code optimization

yzhao062 · Jul 18, 2019 · c2fafbe · c2fafbe
1 parent af8744c
commit c2fafbe
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 7 deletions.
diff --git a/combo/models/classifier_comb.py b/combo/models/classifier_comb.py
@@ -328,8 +328,7 @@ def predict(self, X):
 
         all_scores = np.zeros([X.shape[0], self.len_classifiers_])
 
-        for i in range(self.len_classifiers_):
-            clf = self.classifiers[i]
+        for i, clf in enumerate(self.classifiers):
             if clf.fitted_ != True and self.pre_fitted == False:
                 ValueError('Classifier should be fitted first!')
             else:

diff --git a/combo/test/test_classifier_comb.py b/combo/test/test_classifier_comb.py
@@ -1,6 +1,4 @@
 # -*- coding: utf-8 -*-
-from __future__ import division
-from __future__ import print_function
 
 import os
 import sys

diff --git a/combo/test/test_score_comb.py b/combo/test/test_score_comb.py
@@ -1,8 +1,5 @@
 # -*- coding: utf-8 -*-
 
-from __future__ import division
-from __future__ import print_function
-
 import os
 import sys