Cleans raw data, runs baseline models
Project description
Pywedge
Pywedge is a pip installable Python package that intends to,
-
Quickly preprocess the data by taking the user’s preferred choice of pre-processing techniques & it returns the cleaned datasets to the user in the first step.
-
In the second step, Pywedge offers a baseline class that has a classification summary method & regression summary method, which can return ten various baseline models, which can point the user to explore the best performing baseline model.
Pywedge intends to help the user by quickly preprocessing the data and to rightly point out the best performing baseline model for the given dataset so that the user can spend quality time tuning such a model algorithm.
Pywedge Features
Cleans the raw data frame to fed into ML models. Following data pre_processing will be carried out,
- segregating numeric & categorical columns
- missing values imputation for numeric & categorical columns
- standardization
- feature importance
- class oversampling using SMOTE
- computes 10 different baseline models
Pre_process_data()
Inputs:
- train = train dataframe
- test = test dataframe
- c = any redundant column to be removed (like ID column etc., at present supports a single column removal, subsequent version will provision multiple column removal requirements)
- y = target column name as a string
- type = Classification(Default) / Regression
Returns:
- new_X (cleaned feature columns in dataframe)
- new_y (cleaned target column in dataframe)
- new_test (cleaned stand out test dataset)
!pip install pywedge
import pywedge as pw
ppd = pw.Pre_process_data(train, test, c, y, type='Classification")
new_X, new_y, new_test = ppd.dataframe_clean()
from the image, it can be observed that calling dataframe_clean method does the following,
- Providing a summary of zero & missing values in the training dataset
- Class balance summary
- Categorical column conversion
user is asked for standardization choice...
For binary classification tasks, pywedge computes class balance & asks the user if oversampling using SMOTE to be applied to the data.
baseline_model()
- For classification - classification_summary()
- For Regression - Regression_summary()
Inputs:
- new_x
- new_y
Returns:
Various baseline model metrics
Instantiate the baseline class & call the classification_summary method from baseline_model class,
blm = pw.baseline_model(X,y)
blm.classification_summary()
The classification summary provides Top 10 feature importance (calculated using Adaboost feature importance) and asks for the test size from the user.
The classification summary provides baseline models of 10 different algorithms, user can identify best performing baseline models from the classification summary.
In the same way, regression analysis can be done using a few lines of code.
The following additions to pywedge is planned,
- To handle NLP column
- To handle time series dataset
- To handle stock prices specific analysis
- A separate method to produce good charts
Requires Python 64 bit
THIS IS IN BETA VERSION
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.