Skip to main content

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

  1. 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.
  2. I changed the datatype to be more appropriate.
  3. Do data validation
  4. Mapping / Binning of Categorical Features
  5. LabelEncode a column.
  6. OneHotEncode a column.
  7. 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


Download files

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

Source Distribution

jan883_codebase-0.1.1.tar.gz (1.4 MB view details)

Uploaded Source

Built Distribution

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

jan883_codebase-0.1.1-py3-none-any.whl (1.4 MB view details)

Uploaded Python 3

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

Hashes for jan883_codebase-0.1.1.tar.gz
Algorithm Hash digest
SHA256 0ec802912c6d819ab569244cc57844b2163f5ea81ab405bc26cc2a6e5c392b0a
MD5 b9e7c30c0b9319eade9c788f41e74fdd
BLAKE2b-256 01c589c0709da51f242a4b332df53b50fa4445e387df7078d623531e7242768d

See more details on using hashes here.

File details

Details for the file jan883_codebase-0.1.1-py3-none-any.whl.

File metadata

File hashes

Hashes for jan883_codebase-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 54a8493d4b1a682022f82a8be2a1d60a692b465f907b0856fedf3a3c1d85f2e3
MD5 959e75ac4e7ebc217adbe42dd46f22ce
BLAKE2b-256 05eb5896a450bc8675c6a4e0850f484ff184a876a0127024fc737f6a776bc4aa

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