|
|
@@ -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):
|