Community Notes Data Loading and Preprocessing Pipeline #377
+59
−65
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This module is the backbone of the Community Notes scoring algorithm's data pipeline. It is responsible for the critical first steps of ingesting, cleaning, and preparing the raw data for analysis. Its core functions include:
Data Ingestion: Securely reading and parsing raw TSV files for notes, ratings, note status history, and user enrollment into pandas DataFrames.
Data Cleaning: Handling duplicates, filtering out irrelevant or outdated entries (such as ratings for non-misleading notes), and ensuring data integrity.
Feature Engineering: Creating essential columns used in scoring, such as the unified helpfulNumKey, and identifying high-volume raters.
Data Filtering: Applying core filtering logic based on the minimum number of ratings per note and per rater to create a viable dataset for matrix factorization.
Serialization: Providing utilities to safely write processed data and model outputs to disk, using secure formats like JSON for metadata and TSV for tabular data.
Data Loading Abstraction: Defining the CommunityNotesDataLoader classes which abstract the data source, allowing the scoring algorithm to run seamlessly on either local files or production data sources.
This module ensures that the data fed into the scoring models is clean, consistent, and structured correctly, which is essential for the accuracy and reliability of the entire Community Notes system.