add custom handler and modify log info

Files changed (4) hide show

__pycache__/bert_for_sequence_classification.cpython-37.pyc CHANGED Viewed

Binary files a/__pycache__/bert_for_sequence_classification.cpython-37.pyc and b/__pycache__/bert_for_sequence_classification.cpython-37.pyc differ

__pycache__/bibert_multitask_classification.cpython-37.pyc CHANGED Viewed

Binary files a/__pycache__/bibert_multitask_classification.cpython-37.pyc and b/__pycache__/bibert_multitask_classification.cpython-37.pyc differ

__pycache__/handler.cpython-37.pyc CHANGED Viewed

Binary files a/__pycache__/handler.cpython-37.pyc and b/__pycache__/handler.cpython-37.pyc differ

handler.py CHANGED Viewed

@@ -65,18 +65,18 @@ class EndpointHandler():
     t0 = perf_counter()
     prediction_res = []
-    prediction_p = self.classifier_p(inputs)
-    logger.info(prediction_p)
-    label = prediction_p[0]['label']
-    score = prediction_p[0]['score']
     if label == '0' and score >= 0.75:
       prediction_res = [{"label":"Neutral"}]
     else:
-      classifier_res = self.classifier_s(inputs)
-      logger.info("Prediction %s", classifier_res)
-      label = classifier_res[0]['label']
       for key in idtolabel.keys():
           label = label.replace(key, idtolabel[key])
       prediction_res = [{"label":label}]

     t0 = perf_counter()
     prediction_res = []
+    classifier_pol = self.classifier_p(inputs)
+    label = classifier_pol[0]['label']
+    score = classifier_pol[0]['score']
     if label == '0' and score >= 0.75:
+      logger.info("Prediction polarity %s", classifier_pol)
       prediction_res = [{"label":"Neutral"}]
     else:
+      classifier_subj = self.classifier_s(inputs)
+      logger.info("Prediction subjective %s", classifier_subj)
+      label = classifier_subj[0]['label']
       for key in idtolabel.keys():
           label = label.replace(key, idtolabel[key])
       prediction_res = [{"label":label}]