jsalvad0r commited on
Commit
640b4be
1 Parent(s): f2c2a9e

updated documentation with the Please ask your administrator. flag

Browse files
Files changed (2) hide show
  1. README.md +3 -2
  2. semf1.py +2 -1
README.md CHANGED
@@ -64,8 +64,9 @@ Sem-F1 also accepts multiple optional arguments:
64
  - `tokenize_sentences (bool)`: Flag to indicate whether to tokenize the sentences in the input documents. Default: True.
65
  - `multi_references (bool)`: Flag to indicate whether multiple references are provided. Default: False.
66
  - `gpu (Union[bool, str, int, List[Union[str, int]]])`: Whether to use GPU, CPU or multiple-processes for computation.
67
- - `batch_size (int)`: Batch size for encoding. Default: 32.
68
- - `verbose (bool)`: Flag to indicate verbose output. Default: False.
 
69
 
70
  Refer to the inputs descriptions for more detailed usage as follows:
71
 
 
64
  - `tokenize_sentences (bool)`: Flag to indicate whether to tokenize the sentences in the input documents. Default: True.
65
  - `multi_references (bool)`: Flag to indicate whether multiple references are provided. Default: False.
66
  - `gpu (Union[bool, str, int, List[Union[str, int]]])`: Whether to use GPU, CPU or multiple-processes for computation.
67
+ - `batch_size (int)`: Batch size for encoding. (Default: 32).
68
+ - `verbose (bool)`: Flag to indicate verbose output. (Default: False).
69
+ - `aggregate (bool)`: Flag which is used to automatically compute the mean precision, recall and f1 scores. (Default: False).
70
 
71
  Refer to the inputs descriptions for more detailed usage as follows:
72
 
semf1.py CHANGED
@@ -83,9 +83,10 @@ Args:
83
  List[Union[str, int]] - Multiple GPUs/cpus i.e. use multiple processes when computing embeddings
84
  batch_size (int): Batch size for encoding. Default is 32.
85
  verbose (bool): Flag to indicate verbose output. Default is False.
 
86
 
87
  Returns:
88
- List of Scores dataclass with attributes as follows -
89
  precision: float - precision score
90
  recall: List[float] - List of recall scores corresponding to single/multiple references
91
  f1: float - F1 score (between precision and average recall)
 
83
  List[Union[str, int]] - Multiple GPUs/cpus i.e. use multiple processes when computing embeddings
84
  batch_size (int): Batch size for encoding. Default is 32.
85
  verbose (bool): Flag to indicate verbose output. Default is False.
86
+ aggregate (bool): Flag which is used to automatically compute the mean precision, recall and f1 scores. (Default: False)
87
 
88
  Returns:
89
+ Singleton/List of Scores dataclass with attributes as follows -
90
  precision: float - precision score
91
  recall: List[float] - List of recall scores corresponding to single/multiple references
92
  f1: float - F1 score (between precision and average recall)