Binary classification is backwards

jrc16107 · Apr 2, 2021 · 0e9f2cc · 0e9f2cc
1 parent 5ce41b6
commit 0e9f2cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/__init__.py b/src/__init__.py
@@ -40,7 +40,7 @@ def worker_loop(hostname, model):
 
     # Using the result of the prediction assing a classification
     # TODO - In the future these possibilities should be fetchable from the data service
-    if round(pred) == 1:
+    if round(pred) == 0:
         response_dict['classification'] = "covid-19"
     else:
         response_dict['classification'] = "not covid-19"