modify how to get label

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

@@ -75,10 +75,8 @@ class EndpointHandler():
       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}]
     elapsed = 1000 * (perf_counter() - t0)
     logger.info("Model prediction time: %d ms.", elapsed)

       prediction_res = [{"label":"Neutral"}]
     else:
       classifier_subj = self.classifier_s(inputs)
+      logger.info("Prediction subjective %s", classifier_subj)
+      label = idtolabel.get(classifier_subj[0]['label'])
       prediction_res = [{"label":label}]
     elapsed = 1000 * (perf_counter() - t0)
     logger.info("Model prediction time: %d ms.", elapsed)