chnages in threshold

somasundaram1702 · Aug 17, 2020 · 308a20d · 308a20d
1 parent 7f0a499
commit 308a20d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train_evm.py b/train_evm.py
@@ -87,10 +87,10 @@ def check_accuracy(self):
 
     def save_model(self):
 
-        with open('EVM_model.pkl','wb') as f:
+        with open('./outputs/EVM_model.pkl','wb') as f:
           pickle.dump(self.mevm,f)
 
-        with open('class_names.pkl','wb') as t:
+        with open('./outputs/class_names.pkl','wb') as t:
             pickle.dump(self.uni_classes,t)
 
 if __name__ == '__main__':