Spaces:
Runtime error
Runtime error
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)
|