Update training_SVM.py

RashadGarayev · Mar 11, 2020 · e7e8743 · e7e8743
1 parent 08093b7
commit e7e8743
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/training_SVM.py b/training_SVM.py
@@ -12,7 +12,7 @@
 train_labels = []
 pos_im_path = 'DATAIMAGE/positive/'
 neg_im_path = 'DATAIMAGE/negative/'
-model_path = 'models/model.dat'
+model_path = 'models/models.dat'
 # Load the positive features
 for filename in glob.glob(os.path.join(pos_im_path,"*.png")):
     fd = cv2.imread(filename,0)
@@ -42,4 +42,4 @@
 print('Training...... Support Vector Machine')
 model.fit(train_data,train_labels)
 joblib.dump(model, 'models/models.dat')
-print('Model saved : {}'.format('models/models.dat'))
+print('Model saved : {}'.format('models/models.dat'))