Spaces:
Running
Running
updated documentation with the Please ask your administrator. flag
Browse files
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)
|