Skip to main content

A library for dataset quality checks, preprocessing, and report generation

Project description

HashPrep

Dataset Profiler & Debugger for Machine Learning

[!NOTE] HashPrep is in beta (v0.1.0b3). Core features are fully tested with CI. The API may still evolve based on community feedback.

Overview

HashPrep is a Python library for intelligent dataset profiling and debugging that acts as a comprehensive pre-training quality assurance tool for machine learning projects. Think of it as "Pandas Profiling + PyLint for datasets", designed specifically for machine learning workflows.

It catches critical dataset issues before they derail your ML pipeline, explains the problems, and suggests context-aware fixes.
If you want, HashPrep can even apply those fixes for you automatically.


Features

Key features include:

  • Intelligent Profiling: Detect missing values, skewed distributions, outliers, and data type inconsistencies.
  • ML-Specific Checks: Identify data leakage, dataset drift, class imbalance, and high-cardinality features.
  • Automated Preparation: Get suggestions for encoding, imputation, scaling, and transformations.
  • Rich Reporting: Generate statistical summaries and exportable reports (HTML/PDF/Markdown/JSON) with embedded visualizations.
  • Production-Ready Pipelines: Output reproducible cleaning and preprocessing code (fixes.py) that integrates seamlessly with ML workflows.
  • Modern Themes: Choose between "Minimal" (professional) and "Neubrutalism" (bold) report styles.

Installation

Using pip

pip install hashprep

Using uv (recommended)

# Install uv if you haven't already
curl -LsSf https://astral.sh/uv/install.sh | sh

# Install hashprep
uv pip install hashprep

# Or for development from source
git clone https://github.com/cachevector/hashprep.git
cd hashprep
uv sync

After installation, the hashprep command will be available directly in your terminal.


Usage

HashPrep can be used both as a command-line tool and as a Python library.

CLI Usage

1. Quick Scan

Get a quick summary of critical issues in your terminal.

hashprep scan dataset.csv

Options:

  • --critical-only: Show only critical issues
  • --quiet: Minimal output (counts only)
  • --json: Output in JSON format
  • --target COLUMN: Specify target column for ML-specific checks
  • --checks CHECKS: Run specific checks (comma-separated)
  • --sample-size N: Limit analysis to N rows
  • --no-sample: Disable automatic sampling

Example:

# Scan with target column and specific checks
hashprep scan train.csv --target Survived --checks outliers,high_missing_values,class_imbalance

# Quick scan with JSON output
hashprep scan dataset.csv --json --quiet

2. Detailed Analysis

Get comprehensive details about all detected issues.

hashprep details dataset.csv

Options: Same as scan command

Example:

hashprep details train.csv --target Survived

3. Generate Reports

Generate comprehensive reports in multiple formats with visualizations.

hashprep report dataset.csv --format html --theme minimal

Options:

  • --format {md,json,html,pdf}: Report format (default: md)
  • --theme {minimal,neubrutalism}: HTML report theme (default: minimal)
  • --with-code: Generate Python scripts for fixes and pipelines
  • --full / --no-full: Include/exclude full summaries (default: True)
  • --visualizations / --no-visualizations: Include/exclude plots (default: True)
  • --target COLUMN: Specify target column
  • --checks CHECKS: Run specific checks
  • --comparison FILE: Compare with another dataset for drift detection
  • --sample-size N: Limit analysis to N rows
  • --no-sample: Disable automatic sampling

Examples:

# Generate HTML report with minimal theme
hashprep report dataset.csv --format html --theme minimal --full

# Generate PDF report without visualizations (faster)
hashprep report dataset.csv --format pdf --no-visualizations

# Generate report with automatic fix scripts
hashprep report dataset.csv --with-code

# This creates:
# - dataset_hashprep_report.md (or .html/.pdf/.json)
# - dataset_hashprep_report_fixes.py (pandas script)
# - dataset_hashprep_report_pipeline.py (sklearn pipeline)

# Compare two datasets for drift detection
hashprep report train.csv --comparison test.csv --format html

4. Version

Check HashPrep version.

hashprep version

Available Checks

  • outliers - Detect outliers using z-score
  • duplicates - Find duplicate rows
  • high_missing_values - Columns with high missing data
  • empty_columns - Completely empty columns
  • dataset_missingness - Overall missing data patterns
  • missing_patterns - Correlated missing value patterns
  • high_cardinality - Categorical columns with too many unique values
  • single_value_columns - Constant columns with no variance
  • mixed_data_types - Columns with mixed data types
  • class_imbalance - Imbalanced target variable (requires --target)
  • feature_correlation - Highly correlated numeric features
  • categorical_correlation - Highly associated categorical features
  • mixed_correlation - Numeric-categorical associations
  • data_leakage - Columns identical to target
  • target_leakage_patterns - Features that may leak target information
  • dataset_drift - Distribution drift between datasets (requires --comparison)
  • uniform_distribution - Uniformly distributed numeric columns
  • unique_values - Columns where >95% values are unique
  • high_zero_counts - Columns with excessive zero values
  • skewness - Highly skewed numeric distributions
  • infinite_values - Columns containing infinite values
  • constant_length - String columns with constant character length
  • extreme_text_lengths - Text columns with extreme value lengths
  • datetime_skew - Datetime columns concentrated in one period
  • empty_dataset - Empty or all-missing datasets

Python Library Usage

Basic Analysis

import pandas as pd
from hashprep import DatasetAnalyzer

# Load your dataset
df = pd.read_csv("dataset.csv")

# Create analyzer
analyzer = DatasetAnalyzer(df)

# Run analysis
summary = analyzer.analyze()

# Access results
print(f"Critical issues: {summary['critical_count']}")
print(f"Warnings: {summary['warning_count']}")

# Iterate through issues
for issue in summary['issues']:
    print(f"{issue['severity']}: {issue['description']}")

Analysis with Target Column

# Specify target for ML-specific checks
analyzer = DatasetAnalyzer(
    df,
    target_col='target_column'
)
summary = analyzer.analyze()

Run Specific Checks

# Only run specific checks
analyzer = DatasetAnalyzer(
    df,
    selected_checks=['outliers', 'high_missing_values', 'class_imbalance']
)
summary = analyzer.analyze()

Include Visualizations

# Generate analysis with plots
analyzer = DatasetAnalyzer(df, include_plots=True)
summary = analyzer.analyze()

# Plots are stored in summary['summaries']['plots']

Drift Detection

# Compare two datasets
train_df = pd.read_csv("train.csv")
test_df = pd.read_csv("test.csv")

analyzer = DatasetAnalyzer(
    train_df,
    comparison_df=test_df,
    selected_checks=['dataset_drift']
)
summary = analyzer.analyze()

Generate Reports Programmatically

from hashprep.reports import generate_report

# Analyze dataset
analyzer = DatasetAnalyzer(df, include_plots=True)
summary = analyzer.analyze()

# Generate HTML report
generate_report(
    summary,
    format='html',
    full=True,
    output_file='report.html',
    theme='minimal'
)

# Generate PDF report
generate_report(
    summary,
    format='pdf',
    full=True,
    output_file='report.pdf'
)

# Generate JSON report
generate_report(
    summary,
    format='json',
    full=True,
    output_file='report.json'
)

# Generate Markdown report
generate_report(
    summary,
    format='md',
    full=True,
    output_file='report.md'
)

Generate Fix Scripts

from hashprep.checks.core import Issue
from hashprep.preparers.codegen import CodeGenerator
from hashprep.preparers.pipeline_builder import PipelineBuilder
from hashprep.preparers.suggestions import SuggestionProvider

# After running analysis
analyzer = DatasetAnalyzer(df, target_col='target')
summary = analyzer.analyze()

# Convert issues to proper format
issues = [Issue(**i) for i in summary['issues']]
column_types = summary.get('column_types', {})

# Get suggestions
provider = SuggestionProvider(
    issues=issues,
    column_types=column_types,
    target_col='target'
)
suggestions = provider.get_suggestions()

# Generate pandas fix script
codegen = CodeGenerator(suggestions)
fixes_code = codegen.generate_pandas_script()
with open('fixes.py', 'w') as f:
    f.write(fixes_code)

# Generate sklearn pipeline
builder = PipelineBuilder(suggestions)
pipeline_code = builder.generate_pipeline_code()
with open('pipeline.py', 'w') as f:
    f.write(pipeline_code)

Custom Sampling

from hashprep.utils.sampling import SamplingConfig

# Configure sampling for large datasets
sampling_config = SamplingConfig(max_rows=10000)

analyzer = DatasetAnalyzer(
    df,
    sampling_config=sampling_config,
    auto_sample=True
)
summary = analyzer.analyze()

# Check if sampling occurred
if 'sampling_info' in summary:
    info = summary['sampling_info']
    print(f"Sampled: {info['sample_fraction']*100:.1f}%")

License

This project is licensed under the MIT License.


Contributing

We welcome contributions from the community to make HashPrep better!

Before you get started, please:

  • Review our CONTRIBUTING.md for detailed guidelines and setup instructions
  • Write clean, well-documented code
  • Follow best practices for the stack or component you’re working on
  • Open a pull request (PR) with a clear description of your changes and motivation

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

hashprep-0.1.0b3.tar.gz (73.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

hashprep-0.1.0b3-py3-none-any.whl (88.5 kB view details)

Uploaded Python 3

File details

Details for the file hashprep-0.1.0b3.tar.gz.

File metadata

  • Download URL: hashprep-0.1.0b3.tar.gz
  • Upload date:
  • Size: 73.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.20

File hashes

Hashes for hashprep-0.1.0b3.tar.gz
Algorithm Hash digest
SHA256 189c7fd6a0252372a17d83a09a27bf4110b38e3428a7c6e8a0cb7ccc0c043e52
MD5 142b3c55af1392eb771d623cbede977e
BLAKE2b-256 6e9c85e70eba502783b9212a769023170f0448689883f0ebb47cf8955f06c578

See more details on using hashes here.

File details

Details for the file hashprep-0.1.0b3-py3-none-any.whl.

File metadata

  • Download URL: hashprep-0.1.0b3-py3-none-any.whl
  • Upload date:
  • Size: 88.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.20

File hashes

Hashes for hashprep-0.1.0b3-py3-none-any.whl
Algorithm Hash digest
SHA256 3e6be621065397820f4453c3a8712984d8575cbe8028fd77b47305598cced189
MD5 30dc012b35dadbcc7d2354dcd7449a84
BLAKE2b-256 935165395c02f79b3e3bf922bdbff736d1c010a13333aff5f77d7ddcd36dda9c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page