Taddeus Kroes 14 lat temu
rodzic
commit
1848170ec9
2 zmienionych plików z 3 dodań i 4 usunięć
  1. 1 2
      src/Classifier.py
  2. 2 2
      src/create_classifier.py

+ 1 - 2
src/Classifier.py

@@ -5,8 +5,6 @@ from svmutil import svm_train, svm_problem, svm_parameter, svm_predict, \
 class Classifier:
     def __init__(self, c=None, gamma=None, filename=None, neighbours=3, \
             verbose=0):
-        self.neighbours = neighbours
-
         if filename:
             # If a filename is given, load a model from the given filename
             self.model = svm_load_model(filename)
@@ -19,6 +17,7 @@ class Classifier:
             self.param.gamma = gamma  # Parameter for radial kernel
             self.model = None
 
+        self.neighbours = neighbours
         self.verbose = verbose
 
     def save(self, filename):

+ 2 - 2
src/create_classifier.py

@@ -12,8 +12,8 @@ def load_classifier(neighbours, blur_scale, c=None, gamma=None, verbose=0):
         if verbose:
             print 'Loading classifier...'
 
-        classifier = Classifier(filename=classifier_file, verbose=verbose)
-        classifier.neighbours = neighbours
+        classifier = Classifier(filename=classifier_file, \
+                neighbours=neighbours, verbose=verbose)
     elif c != None and gamma != None:
         if verbose:
             print 'Training new classifier...'