Personal codebase for data science and machine learning projects. Includes data preprocessing, feature engineering, model selection, and model evaluation.
Project description
jan883-codebase - EDA and Model Selection Toolkit
This repository contains a collection of Python functions designed to streamline Exploratory Data Analysis (EDA) and model selection processes. The toolkit is divided into three main sections: EDA Level 1, EDA Level 2, and Model Selection, each providing a set of utility functions to assist in data transformation, analysis, and model evaluation.
Package Content
EDA Level 1 — Transformation of Original Data
This section focuses on the initial transformation and cleaning of raw data. Key functionalities include:
- Column Name Standardization: Convert column names to lowercase and replace spaces with underscores.
- Handling Missing Values: Fill null/NaN values with appropriate data.
- Data Type Conversion: Ensure columns have the correct data types.
- Data Validation: Validate the integrity of the dataset.
- Categorical Feature Mapping/Binning: Map or bin categorical features for better analysis.
- Encoding: Perform Label Encoding and One-Hot Encoding on categorical columns.
- Outlier Removal: Remove outliers using Z-score.
- Imputation: Impute missing values using strategies like median, mean, or mode.
EDA Level 2 — Understanding of Transformed Data
This section delves deeper into understanding the transformed data through advanced analysis techniques:
- Correlation Analysis: Generate correlation heatmaps and identify maximum pairwise correlations.
- IV/WOE Analysis: Calculate Information Value (IV) and Weight of Evidence (WOE) to assess feature predictive power.
- Feature Importance: Extract feature importance from models.
- Statistical Tests: Perform individual t-tests for classification and regression problems.
- QQ Plots: Create QQ plots for feature comparison.
- Data Scaling: Scale data for exploration or model training.
- Train-Test Splitting: Split data into training, validation, and test sets.
- Oversampling: Use SMOTE to oversample minority classes.
Model Selection — Choosing the Right Model
This section provides tools for evaluating and selecting the best models for your data:
- Feature Importance Visualization: Plot feature importance for a given model.
- Learning Curves: Generate learning curves for regression and classification models.
- Recursive Feature Elimination: Perform recursive feature elimination with cross-validation.
- Model Evaluation: Evaluate classification and regression models using key metrics.
- Best Model Selection: Test multiple regression and classification models to identify the best-performing one.
- Clustering Analysis: Use the Elbow Method, Intercluster Distance, and Silhouette Visualizer to determine optimal clustering parameters.
Key Features
- Custom Functions: Pre-built functions for common EDA and model selection tasks.
- Modular Design: Functions are designed to be reusable and adaptable to different datasets.
- Visualization: Integrated visualization tools for better data understanding and model evaluation.
This toolkit is ideal for data scientists and analysts looking to accelerate their EDA and model selection workflows. Whether you're working on classification, regression, or clustering tasks, this repository provides the tools to make your process more efficient and insightful.
Data Pre-processing
from jan883_codebase.data_preprocessing.eda import *
# Run this function for a printout of included functions in Jupyter Notebook.
eda0()
eda1()
eda2()
EDA Level 0 - Pure Understanding of Original Data
Basic check on the column datatype, null counts, distinct values, to get a better understanding of the data. I also created a distinct values count dictionary where I go the top 10 counts and their distinct values displayed so I could roughly gauge how significant the distinct values are in the dataset. Custom Functions
inspect_df(df) Run df.head(), df.describe(), df.isna().sum() & df.duplicated().sum() on your dataframe.column_summary(df)Create a dataframe with column info, dtype, value_counts, etc.column_summary_plus(df)Create a dataframe with column info, dtype, value_counts, plus df.decsribe() info.univariate_analysis(df)Perform Univariate Analysis of numeric columns.
EDA Level 1 — Transformation of Original Data
- I changed the column names to all be in small letters and spaces to be changed to underscore. I also changed it to names that I feel are more generic and categorized, for easy interpretation. 2. I filled in the empty null / NaN rows with values I feel make sense.
- I changed the datatype to be more appropriate.
- Do data validation
- Mapping / Binning of Categorical Features
- LabelEncode a column.
- OneHotEncode a column.
- Impute missing values Custom Functions
update_column_names(df)Update Column names, replace " " with "_".label_encode_column(df, col_name)Label encode a df column returing a df with the new column (original col dropped).one_hot_encode_column(df, col_name)One Hot Encode a df column returing a df with the new column (original col dropped).train_no_outliers = remove_outliers_zscore(train, threshold=3)Remove outliers using Z score.df_imputed = impute_missing_values(df, strategy='median')Impute missing values in DF
EDA Level 2 — Understanding of Transformed Data
- Correlation Analysis
- IV / WOE Values - Information Value (IV) quantifies the prediction power of a feature. Short story is, we are looking for IV of 0.1 to 0.5. <0.1 weak, >0.5 to good to be true.
- Feature Importance from Models
- Statistical Tests
- Create QQ Plots - Further Data Analysis on Imputed Data
- scale_df(X) should be used in exploration as it does not scale X_test.
- scale_X_train_X_test(X_train, X_test, scaler="standard", save_scaler=False) is a full scaler solution, scales X_test (fittransform) and saves the scaler to disk. Custom Functions
correlation_analysis(df, width=16, height=12)Correlation Heatmap & Maximum pairwise correlation.newDF, woeDF = iv_woe(df, target, bins=10, show_woe=False)Returns newDF, woeDF. IV / WOE Values - Information Value (IV) quantifies the prediction power of a feature. We are looking for IV of 0.1 to 0.5. For those with IV of 0, there is a high chance it is the way it is due to imbalance of data, resulting in lack of binning. Keep this in mind during further analysis.individual_t_test_classification(df, y_column, y_value_1, y_value_2, list_of_features, alpha_val=0.05, sample_frac=1.0, random_state=None)Statistical test of individual features - Classification problem.individual_t_test_regression(df, y_column, list_of_features, alpha_val=0.05, sample_frac=1.0, random_state=None)Statistical test of individual features - Regressions problem.create_qq_plots(df, reference_col)Create QQ plots of the features in a dataframe.volcano_plot(df, reference_col)Create Volcano Plot with P-values.X, y = define_X_y(df, target)Define X and y..X_train, X_test, y_train, y_test = train_test_split_custom(X, y, test_size=0.2, random_state=42)Split train, test.X_train, X_val, X_test, y_train, y_val, y_test = train_val_test_split(X, y, val_size=0.2, test_size=0.2, random_state=42)Split train, val, test.X_train_res, y_train_res = oversample_SMOTE(X_train, y_train, sampling_strategy="auto", k_neighbors=5, random_state=42)Oversample minority class.scaled_X = scale_df(X, scaler='standard')only scales X, does not scale X_test or X_val.scaled_X_train, scaled_X_test = scale_X_train_X_test(X_train, X_test, scaler="standard", save_scaler=False)Standard, MinMax and Robust Scaler. X_train uses fit_transform, X_test uses transform.sample_df(df, n_samples)Take a sample of the full df.
Model Selection
from jan883_codebase.data_preprocessing.eda import *
ml0() # Run this function for a printout of included functions in Jupyter Notebook.
- Feature Importance Plot
- Evaluate Classification Model
- Evaluate Regression Model
- Test Regression Models
- Test Classification Modeks Custom Functions
feature_importance_plot(model, X, y)Plot Feature Importance using a single model.evaluate_classification_model(model, X, y, cv=5)Plot peformance metrics of single classification model.evaluate_regression_model(model, X, y)Plot peformance metrics of single regression model.test_regression_models(X, y, test_size=0.2, random_state=None, scale_data=False)Test Regression models.test_classification_models(X, y, test_size=0.2, random_state=None, scale_data=False)Test Classification models.
NotionHelper class
import os
# Set the environment variable
os.environ["NOTION_TOKEN"] = "<your-notion-token>"
from jan883_codebase.notion_api.notionhelper import NotionHelper
nh = NotionHelper() # Instantiate the class
nh.get_all_pages_as_dataframe(database_id)
A helper class to interact with the Notion API.
Methods
authenticate(): Authenticates with the Notion API using a token from environment variables.
get_database(database_id): Fetches the schema of a Notion database given its database_id.
notion_search_db(database_id, query=""): Searches for pages in a Notion database that contain the specified query in their title.
notion_get_page(page_id): Returns the JSON of the page properties and an array of blocks on a Notion page given its page_id.
create_database(parent_page_id, database_title, properties): Creates a new database in Notion under the specified parent page with the given title and properties.
new_page_to_db(database_id, page_properties): Adds a new page to a Notion database with the specified properties.
append_page_body(page_id, blocks): Appends blocks of text to the body of a Notion page.
get_all_page_ids(database_id): Returns the IDs of all pages in a given Notion database.
get_all_pages_as_json(database_id, limit=None): Returns a list of JSON objects representing all pages in the given database, with all properties.
get_all_pages_as_dataframe(database_id, limit=None): Returns a Pandas DataFrame representing all pages in the given database, with selected properties.
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.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file jan883_codebase-0.1.1.tar.gz.
File metadata
- Download URL: jan883_codebase-0.1.1.tar.gz
- Upload date:
- Size: 1.4 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.10.6
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0ec802912c6d819ab569244cc57844b2163f5ea81ab405bc26cc2a6e5c392b0a
|
|
| MD5 |
b9e7c30c0b9319eade9c788f41e74fdd
|
|
| BLAKE2b-256 |
01c589c0709da51f242a4b332df53b50fa4445e387df7078d623531e7242768d
|
File details
Details for the file jan883_codebase-0.1.1-py3-none-any.whl.
File metadata
- Download URL: jan883_codebase-0.1.1-py3-none-any.whl
- Upload date:
- Size: 1.4 MB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.0.1 CPython/3.10.6
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
54a8493d4b1a682022f82a8be2a1d60a692b465f907b0856fedf3a3c1d85f2e3
|
|
| MD5 |
959e75ac4e7ebc217adbe42dd46f22ce
|
|
| BLAKE2b-256 |
05eb5896a450bc8675c6a4e0850f484ff184a876a0127024fc737f6a776bc4aa
|