Patch default dataset name in views to german
diff --git a/api/views.py b/api/views.py
index 2591710..85c2f8a 100644
--- a/api/views.py
+++ b/api/views.py
@@ -87,7 +87,7 @@
try:
classifier = self.request.query_params.get("classifier")
- region = self.request.query_params.get("dataset", "Germany")
+ region = self.request.query_params.get("dataset", "german")
version = self.request.query_params.get("version", "0.0.1")
status = self.request.query_params.get("status", "production")
diff --git a/ml/classifiers.py b/ml/classifiers.py
index 3b894d0..bc12a1f 100644
--- a/ml/classifiers.py
+++ b/ml/classifiers.py
@@ -111,7 +111,6 @@
data: array
Data to perform prediction on.
"""
- log.info("------------Predict Probability------------")
return self.model.predict_proba(data)
def postprocessing(self, prediction):
@@ -121,7 +120,6 @@
return {"probability": prediction[1], "label": label}
def compute_prediction(self, data: Dict[str, Any]):
- log.info("------------Compute Prediction------------")
try:
input_data = self.preprocessing(data)
prediction = self.predict(input_data)[0]