svm.py 文件源码

def train(self):
        """
            Train SVM

        """
        print "Starting Training"

        rs = ShuffleSplit(self.length,n_iter=self.args.fold,test_size=self.args.test_size,random_state=self.args.random_state)
        self.fold = 1
        for train_index,test_index in rs:
            self.train_images,self.train_labels = self.images[train_index,...],self.labels[train_index,...]
            self.valid_images,self.valid_labels = self.images[test_index,...],self.labels[test_index,...]
            #pdb.set_trace()

            self.svm_classifier = self.classifier.fit(self.train_images,self.train_labels)
            self.test(self.valid_images)
            self.fold+=1