diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/sec_certs/model/references_nlp/segment_extractor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/sec_certs/model/references_nlp/segment_extractor.py b/src/sec_certs/model/references_nlp/segment_extractor.py index b30d5db7..d63208ec 100644 --- a/src/sec_certs/model/references_nlp/segment_extractor.py +++ b/src/sec_certs/model/references_nlp/segment_extractor.py @@ -34,7 +34,7 @@ def swap_and_filter_dict(dct: dict[str, Any], filter_to_keys: set[str]): return {key: frozenset(val) for key, val in new_dct.items() if key in filter_to_keys} -def fill_reference_segments(record: ReferenceRecord, n_sent_before: int = 4, n_sent_after: int = 4) -> ReferenceRecord: +def fill_reference_segments(record: ReferenceRecord, n_sent_before: int = 2, n_sent_after: int = 1) -> ReferenceRecord: """ Compute indices of the sentences containing the reference keyword, take their surrounding sentences and join them. """ |