PreProcessor
Use the PreProcessor to normalize white spaces, get rid of headers and footers, clean empty lines in your Documents, or split them into smaller pieces. PreProcessor is useful in an indexing pipeline to prepare your files for search.
Splitting is generally recommended for long Documents as it makes the Retriever's job easier and speeds up Question Answering. For suggestions on how best to split your documents, see Optimization.
Tutorial
To start working with code examples, have a look at the How to Preprocess Documents tutorial. For ideas on what you can do at indexing time, see the DocumentClassifier at Index Time tutorial.
Usage
To initialize PreProcessor
, run:
from haystack.nodes import PreProcessor
processor = PreProcessor(
clean_empty_lines=True,
clean_whitespace=True,
clean_header_footer=True,
remove_substrings=None,
split_by="word",
split_length=200,
split_respect_sentence_boundary=True,
split_overlap=0,
max_chars_check: int = 10_000
)
Argument | Type | Description |
---|---|---|
clean_empty_lines | bool | Normalizes 3 or more consecutive empty lines to be just two empty lines. |
clean_whitespace | bool | Removes any whitespace at the beginning or end of each line in the text. |
clean_header_footer | bool | Removes any long header or footer texts that are repeated on each page. |
remove_substrings | list | Remove specified substrings from the text. If no value is provided, an empty list is created by default. |
split_by | string | Determines what unit the document is split by. Choose from 'word' , 'sentence' or 'passage' . |
split_length | int | Sets a maximum number of 'word' , 'sentence' or 'passage' units per output document |
split_respect_sentence_boundary | bool | Ensures that document boundaries do not fall in the middle of sentences |
split_overlap | int | Sets the amount of overlap between two adjacent documents after a split. Setting this to a positive number essentially enables the sliding window approach. |
max_chars_check | int | Sets the maximum length of a document. If the document exceeds this limit, it will generate a warning and be split at the maximum character value. The resulting fragments will be further cut if they still exceed the maximum limit. |
There are also additional parameters you can set to customize your PreProcessing:
Argument | Type | Description |
---|---|---|
tokenizer_model_folder | path | Path to the folder containing the NTLK PunktSentenceTokenizer models, if loading a model from a local path. Leave empty otherwise. |
language | str | The language used by "nltk.tokenize.sent_tokenize" in iso639 format. |
id_hash_keys | list | Generate the document id from a custom list of strings that refer to the document's attributes. |
progress_bar | bool | Enable or disable the progress bar. |
add_page_number | bool | Add the number of the page a paragraph occurs in to the Document's meta field "page". Page boundaries are determined by "\f" character which is added in between pages by PDFToTextConverter, TikaConverter, ParsrConverter and AzureConverter. |
See the complete list of parameters with descriptions in PreProcessor API documentation.
To run the PreProcessor
by itself, run:
doc = converter.convert(file_path=file, meta=None)
docs = processor.process(doc)
To use PreProcessor
in a pipeline, run:
from haystack.pipelines import Pipeline
from haystack.nodes import PreProcessor, TextConverter, Retriever
from haystack.nodes import DeepsetCloudDocumentStore
pipeline = Pipeline()
pipeline.add_node(component=text_converter, name="TextConverter", inputs=["File"])
pipeline.add_node(component=preprocessor, name="PreProcessor", inputs=["TextConverter"])
pipeline.add_node(component=retriever, name="EmbeddingRetriever", inputs=["PreProcessor"])
pipeline.add_node(component=document_store, name="DeepsetCloudDocumentStore", inputs="EmbeddingRetriever")
Document Format
When you are not using an indexing Pipeline, the PreProcessor can take either Document
objects (recommended) as input or plain dictionaries. To learn more about the Document
class, see Documents, Answers, and Labels.
# Option 1: Native Haystack Documents
docs = [
Document(
content='DOCUMENT_TEXT_HERE',
meta={'name': DOCUMENT_NAME, ...}
...
), ...
]
# Option 2: Plain dictionary
docs = [
{
'content': 'DOCUMENT_TEXT_HERE',
'meta': {'name': DOCUMENT_NAME, ...}
}, ...
]
Updated over 1 year ago