Ikala-allen commited on
Commit
2451a52
1 Parent(s): ff3eea6

add default mode = "strict "

Browse files
Files changed (1) hide show
  1. relation_extraction.py +1 -1
relation_extraction.py CHANGED
@@ -132,7 +132,7 @@ class relation_extraction(evaluate.Metric):
132
  def _download_and_prepare(self, dl_manager):
133
  pass
134
 
135
- def _compute(self, predictions, references, mode, detailed_scores=False, relation_types=[]):
136
  """
137
  This method computes and returns various scoring metrics for the prediction model based on the mode specified, including Precision, Recall, F1-Score and others. It evaluates the model's predictions against the provided reference data.
138
 
 
132
  def _download_and_prepare(self, dl_manager):
133
  pass
134
 
135
+ def _compute(self, predictions, references, mode="strict", detailed_scores=False, relation_types=[]):
136
  """
137
  This method computes and returns various scoring metrics for the prediction model based on the mode specified, including Precision, Recall, F1-Score and others. It evaluates the model's predictions against the provided reference data.
138