use diffrent svm model

ipranjal · Dec 2, 2023 · 6aa726c · 6aa726c
1 parent d22f21e
commit 6aa726c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/app.py b/api/app.py
@@ -8,7 +8,7 @@
 
 def load_model(model_name):
     dirname = os.path.dirname(__file__)
-    filename_svm = os.path.join(dirname, '../models/svmgamma:0.001_C:1.joblib')
+    filename_svm = os.path.join(dirname, '../models/svmgamma:0.001_C:2.joblib')
     svm = load(filename_svm)
     filename_tree = os.path.join(dirname, '../models/treemax_depth:100.joblib')
     tree = load(filename_tree)

diff --git a/test_api.py b/test_api.py
@@ -23,9 +23,9 @@ def get_processed_data():
 
 def test_predict_svm():
     X,y = get_processed_data()
-    response = app.test_client().post("/predict/svm", json={"image":X[3].tolist()})
+    response = app.test_client().post("/predict/svm", json={"image":X[9].tolist()})
     assert response.status_code == 200    
-    assert response.get_json()['prediction'] == [3]
+    assert response.get_json()['prediction'] == [9]
 
 def test_predict_tree():
     X,y = get_processed_data()