Update README.md

README.md

@@ -47,7 +47,7 @@ evaluation_scores = module.compute(predictions=predictions, references=reference
 - **predictions** (`list` of `list` of `dictionary`): a list of list of dictionary with every element's relation_type and their entity name
 - **references** (`list` of `list` of `dictionary`): a list of list of dictionary with every element's relation_type and their entity name
 - **mode** (`str`): define strict or boundaries mode for evaluation, strict mode consider "head_type" and "tail_type", boundaries mode doesn't consider "head_type" and "tail_type"
-- **only** (`bool`): True for only output all relation_type score. False for output every relation_type score, default True
+- **show** (`bool`): False for only output all relation_type score. True for output every relation_type score, default False
 - **relation_types** (`list`): define which relation type that need to be evaluate and show, if not given, it will construct relation_types from ground truth, default []
  
 ### Output Values
@@ -79,7 +79,7 @@ Output Example:
 Remind : Macro_f1、Macro_p、Macro_r、p、r、f1 are always a number between 0 and 1. And tp、fp、fn depend on how many data inputs.
 
 ### Examples
-Example1 : only one prediction and reference, mode=strict, only output total relation score
+Example1 : only one prediction and reference, mode=strict, show=False, only output total relation score
 ```python
 metric_path = "Ikala-allen/relation_extraction"
 module = evaluate.load(metric_path)
@@ -96,12 +96,12 @@ predictions = [
     {"head": "tinadaviespigments", "head_type": "brand", "type": "sell", "tail": "國際認證之色乳", "tail_type": "product"},
   ]
 ]
-evaluation_scores = module.compute(predictions=predictions, references=references, mode="strict", only=True, relation_types=[])
+evaluation_scores = module.compute(predictions=predictions, references=references, mode="strict", show=False, relation_types=[])
 print(evaluation_scores)
 >>> {'tp': 1, 'fp': 1, 'fn': 2, 'p': 50.0, 'r': 33.333333333333336, 'f1': 40.0, 'Macro_f1': 25.0, 'Macro_p': 25.0, 'Macro_r': 25.0}
 ```
 
-Example2 : only one prediction and reference, mode=boundaries, only output total relation score
+Example2 : only one prediction and reference, mode=boundaries, show=False, only output total relation score
 ```python
 metric_path = "Ikala-allen/relation_extraction"
 module = evaluate.load(metric_path)
@@ -118,12 +118,12 @@ predictions = [
     {"head": "tinadaviespigments", "head_type": "brand", "type": "sell", "tail": "國際認證之色乳", "tail_type": "product"},
   ]
 ]
-evaluation_scores = module.compute(predictions=predictions, references=references, mode="strict", only=True, relation_types=[])
+evaluation_scores = module.compute(predictions=predictions, references=references, mode="strict", show=False, relation_types=[])
 print(evaluation_scores)
->>> {'tp': 2, 'fp': 0, 'fn': 1, 'p': 100.0, 'r': 66.66666666666667, 'f1': 80.0, 'Macro_f1': 50.0, 'Macro_p': 50.0, 'Macro_r': 50.0}
+>>> {'tp': 1, 'fp': 1, 'fn': 2, 'p': 50.0, 'r': 33.333333333333336, 'f1': 40.0, 'Macro_f1': 25.0, 'Macro_p': 25.0, 'Macro_r': 25.0}
 ```
 
-Example3 : two or more prediction and reference, mode=boundaries, only=False, output all relation type score
+Example3 : two or more prediction and reference, mode=boundaries, show=True, output all relation type score
 ```python
 metric_path = "Ikala-allen/relation_extraction"
 module = evaluate.load(metric_path)
@@ -147,12 +147,12 @@ predictions = [
     {'head': 'SNTAIWAN', 'tail': '大馬士革玫瑰有機光燦系列', 'head_type': 'brand', 'tail_type': 'product', 'type': 'sell'}  
   ]
 ]
-evaluation_scores = module.compute(predictions=predictions, references=references, mode="boundaries", only=False, relation_types=[])
+evaluation_scores = module.compute(predictions=predictions, references=references, mode="boundaries", show=True, relation_types=[])
 print(evaluation_scores)
 >>> {'sell': {'tp': 3, 'fp': 1, 'fn': 0, 'p': 75.0, 'r': 100.0, 'f1': 85.71428571428571}, 'belongs_to': {'tp': 0, 'fp': 0, 'fn': 1, 'p': 0, 'r': 0, 'f1': 0}, 'ALL': {'tp': 3, 'fp': 1, 'fn': 1, 'p': 75.0, 'r': 75.0, 'f1': 75.0, 'Macro_f1': 42.857142857142854, 'Macro_p': 37.5, 'Macro_r': 50.0}}
 ```
 
-Example 4 : two or more prediction and reference, mode=boundaries, only=False, only output total relation score, relation_types  = ["belongs_to"], only consider belongs_to type score
+Example 4 : two or more prediction and reference, mode=boundaries, show=True, output all relation type score, relation_types  = ["belongs_to"], only consider belongs_to type score
 ```python
 metric_path = "Ikala-allen/relation_extraction"
 module = evaluate.load(metric_path)
@@ -176,7 +176,7 @@ predictions = [
     {'head': 'SNTAIWAN', 'tail': '大馬士革玫瑰有機光燦系列', 'head_type': 'brand', 'tail_type': 'product', 'type': 'sell'}  
   ]
 ]
-evaluation_scores = module.compute(predictions=predictions, references=references, mode="boundaries", only=False, relation_types=["belongs_to"])
+evaluation_scores = module.compute(predictions=predictions, references=references, mode="boundaries", show=True, relation_types=["belongs_to"])
 print(evaluation_scores)  
 >>> {'belongs_to': {'tp': 0, 'fp': 0, 'fn': 1, 'p': 0, 'r': 0, 'f1': 0}, 'ALL': {'tp': 0, 'fp': 0, 'fn': 1, 'p': 0, 'r': 0, 'f1': 0, 'Macro_f1': 0.0, 'Macro_p': 0.0, 'Macro_r': 0.0}}
 ```