Text Cleaning

Remove undesirable text such as improperly decoded Unicode characters, inconsistent line spacing, or excessive URLs from documents being pre-processed for your dataset using NeMo Curator.

One common issue in text datasets is improper Unicode character encoding, which can result in garbled or unreadable text, particularly with special characters like apostrophes, quotes, or diacritical marks. For example, the input sentence "The Mona Lisa doesn't have eyebrows." from a given document may not have included a properly encoded apostrophe ('), resulting in the sentence decoding as "The Mona Lisa doesn’t have eyebrows.".

NeMo Curator enables you to easily run this document through the default UnicodeReformatter module to detect and remove the unwanted text, or you can define your own custom Unicode text cleaner tailored to your needs.

How it Works

NeMo Curator provides the following modules for cleaning text:

• UnicodeReformatter: Uses ftfy to fix broken Unicode characters. Modifies the “text” field of the dataset by default. The module accepts extensive configuration options for fine-tuning Unicode repair behavior. Please see the ftfy documentation for more information about parameters used by the UnicodeReformatter.
• NewlineNormalizer: Uses regex to replace 3 or more consecutive newline characters in each document with only 2 newline characters.
• UrlRemover: Uses regex to remove all URLs in each document.

You can use these modules individually or sequentially in a cleaning pipeline.

Usage

1
from nemo_curator.core.client import RayClient
2
from nemo_curator.pipeline import Pipeline
3
from nemo_curator.stages.text.io.reader import JsonlReader
4
from nemo_curator.stages.text.io.writer import JsonlWriter
5
from nemo_curator.stages.text.modifiers import UnicodeReformatter, UrlRemover, NewlineNormalizer
6
from nemo_curator.stages.text.modules import Modify
7
8
def main():
9
    # Initialize Ray client
10
    ray_client = RayClient()
11
    ray_client.start()
12
13
    # Create processing pipeline
14
    pipeline = Pipeline(
15
        name="text_cleaning_pipeline",
16
        description="Clean text data using Unicode reformatter, newline normalizer, and URL remover"
17
    )
18
    
19
    # Add reader stage
20
    pipeline.add_stage(JsonlReader(file_paths="books/"))
21
    
22
    # Add processing stages
23
    pipeline.add_stage(Modify(UnicodeReformatter()))
24
    pipeline.add_stage(Modify(NewlineNormalizer()))
25
    pipeline.add_stage(Modify(UrlRemover()))
26
    
27
    # Add writer stage
28
    pipeline.add_stage(JsonlWriter(path="cleaned_books/"))
29
30
    # Execute pipeline
31
    results = pipeline.run()
32
33
    # Stop Ray client
34
    ray_client.stop()
35
    
36
if __name__ == "__main__":
37
    main()

Custom Text Cleaner

You can create your own custom text cleaner by extending the DocumentModifier class. The implementation of UrlRemover demonstrates this approach:

1
import re
2
3
from nemo_curator.stages.text.modifiers.doc_modifier import DocumentModifier
4
5
URL_REGEX = re.compile(r"https?://\S+|www\.\S+", flags=re.IGNORECASE)
6
7
class UrlRemover(DocumentModifier):
8
    """
9
    Removes all URLs in a document.
10
    """
11
12
    def __init__(self):
13
        super().__init__()
14
15
    def modify_document(self, text: str) -> str:
16
        return URL_REGEX.sub("", text)

To create a custom text cleaner, inherit from the DocumentModifier class and implement the constructor and modify_document method.