Clean your data using a scikit-learn transformer in a single line of code
Project description
pandas_dq
Analyze and clean your data in a single line of code with a Scikit-Learn compatible Transformer.
Table of Contents
- What is pandas_dq
- How to use pandas_dq
- How to install pandas_dq
- Usage
- API
- Maintainers
- Contributing
- License
Introduction
What is pandas_dq?
pandas_dq
is a new python library for automatically cleaning your dirty dataset using pandas scikit_learn functions. You can analyze your dataset and fix them - all in a single line of code!
Uses
pandas_dq
has two important modules: find_dq
and Fix_DQ
.
1. find_dq function
`find_dq` is a function that is probably the most popular way to use pandas_dq and it performs following data quality analysis steps:
- It detects missing values and suggests to impute them with mean, median, mode, or a constant value.
- It identifies rare categories and suggests to group them into a single category or drop them.
- It finds infinite values and suggests to replace them with NaN or a large value.
- It detects mixed data types and suggests to convert them to a single type or split them into multiple columns.
- It detects outliers and suggests to remove them or use robust statistics.
- It detects high cardinality features and suggests to reduce them using encoding techniques or feature selection methods.
- It detects highly correlated features and suggests to drop one of them or use dimensionality reduction techniques.
- It detects duplicate rows and columns and suggests to drop them or keep only one copy.
- It detects skewed distributions and suggests to apply transformations or scaling techniques.
- It detects imbalanced classes and suggests to use resampling techniques or class weights.
- It detects feature leakage and suggests to avoid using features that are not available at prediction time.
2. Fix_DQ class: a scikit_learn transformer which can detect data quality issues and clean them all in one line of code
`Fix_DQ` is a great way to clean an entire train data set and apply the same steps in an MLOps pipeline to a test dataset. `Fix_DQ` can be used to detect most issues in your data (similar to find_dq but without the target related steps) in one step (during `fit` method). This transformer can then be saved (or "pickled") for applying the same steps on test data either at the same time or later.
How can we use Fix_DQ in GridSearchCV to find the best model pipeline?
This is another way to find the best data cleaning steps for your train data and then use the cleaned data in hyper parameter tuning using GridSearchCV or RandomizedSearchCV along with a LightGBM or an XGBoost or a scikit-learn model.
Install
Prerequsites:
- pandas_dq is built using pandas, numpy and scikit-learn - that's all. It should run on almost all Python3 Anaconda installations without additional installs. You won't have to import any special libraries.
pip install pandas_dq
To install from source:
cd <pandas_dq_Destination>
git clone git@github.com:AutoViML/pandas_dq.git
or download and unzip https://github.com/AutoViML/pandas_dq/archive/master.zip
conda create -n <your_env_name> python=3.7 anaconda
conda activate <your_env_name> # ON WINDOWS: `source activate <your_env_name>`
cd pandas_dq
pip install -r requirements.txt
Usage
You can invoke `Fix_DQ` as a scikit-learn compatible fit and transform object. See syntax below.
from pandas_dq import Fix_DQ
# Call the transformer to print data quality issues
# as well as clean your data - all in one step
# Create an instance of the fix_data_quality transformer with default parameters
fdq = Fix_DQ()
# Fit the transformer on X_train and transform it
X_train_transformed = fdq.fit_transform(X_train)
# Transform X_test using the fitted transformer
X_test_transformed = fdq.transform(X_test)
if you are not using the Transformer, you can simply call the function, find_dq
from pandas_dq import find_dq
find_dq(df, target=target, verbose=0)
API
pandas_dq has a very simple API with the following inputs. You need to create a sklearn-compatible transformer pipeline object by importing Fix_DQ from pandas_dq library.
Once you import it, you can define the object by giving several options such as:
Arguments
Caution: X_train and y_train must be pandas Dataframes or pandas Series. I have not tested it on numpy arrays. You can try your luck.
find_dq
has only 3 arguments:
df
: default is a pandas DataFrame.target
: default:None
. Otherwise, it should be a string name representing the name of a column in df. You can leave it asNone
if you don't want any target related issues.verbose
: This has 2 possible states:0
silent output. Great for running where it prints only high level data quality issues.1
more verbiage. Great for knowing details behind each issue and what the suggestions are.
Fix_DQ
has slightly more arguments:
quantile
: float (0.75): Define a threshold for IQR for outlier detection. Could be any float between 0 and 1. If quantile is set toNone
, then no outlier detection will take place.cat_fill_value
: string ("missing"): Define a fill value for missing categories in your object or categorical variables. This is a global default for your entire dataset. I will try to change it to a dictionary so that you can specify different values for different columns.num_fill_value
: integer or float (999): Define a fill value for missing numbers in your integer or float variables. This is a global default for your entire dataset. I will try to change it to a dictionary so that you can specify different values for different columns.rare_threshold
: float (0.05): Define a threshold for rare categories. If a certain category in a column is less 5% (say) of samples, then it will considered rare. All rare categories will be merged with a category value called "Rare".correlation_threshold
: float (0.8): Define a correlation limit. Anything above this limit, the variable will be dropped.
Maintainers
Contributing
PRs accepted.
License
Apache License 2.0 © 2020 Ram Seshadri
Note of Gratitude
This libray would not have been possible without the help of ChatGPT and Bard. This library is dedicated to the thousands of people who worked to create LLM's.
DISCLAIMER
This project is not an official Google project. It is not supported by Google and Google specifically disclaims all warranties as to its quality, merchantability, or fitness for a particular purpose.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.