Skip to content

ActurialCapital/blocks

Repository files navigation

logo

Overview
Open Source BSD 3-clause
Code !pypi !python-versions
CI/CD !codecov
Downloads PyPI - Downloads PyPI - Downloads Downloads

Table of Contents
  1. About The Project
  2. Installation
  3. Testing
  4. Getting Started
  5. Contributing
  6. License

About The Project

Introduction

blocks is a package designed to extend the functionality of scikit-learn by providing additional blocks for creating custom pipelines, easy-to-use base transformers, and useful decorators. This package aims to simplify the process of building and managing machine learning workflows in Python.

The current version of the package offers:

  • Custom Pipelines: Easily create and manage custom pipelines
  • Base Transformers and Samplers: A collection of base transformers and samplers to streamline feature transformation
  • Decorators: Handy decorators to simplify repetitive tasks

(back to top)

Built With

  • scikit-learn = "^1.5.0"
  • imbalanced-learn = "^0.12.3"
  • pandas = "^2.2.2"
  • numpy = "^1.26.4"

(back to top)

Installation

The easiest way to install blocks is via PyPI:

pip install python-blocks

Or via poetry:

poetry add python-blocks

Testing

To run the test suite after installation, follow these steps from the source directory. First, install pytest version 8.2.2:

pip install pytest==8.2.2

Then run pytest as follow:

pytest tests

Alternatively, if you are using poetry, execute:

poetry run pytest

For more information, visit our Codecov page.

(back to top)

Getting Started

Pipeline

  • Callback function that logs information in between each intermediate step
  • Access particular named step data
  • Inherites from imblearn pipeline, which works with both transformers and samplers

Dataset

>>> from sklearn.datasets import make_regression
>>> X, y = make_regression(n_samples=1000, n_features=10, random_state=42)

Model with both recorded and logged callbacks

>>> from sklearn.preprocessing import StandardScaler
>>> from sklearn.linear_model import LinearRegression
>>> from sklego.meta import EstimatorTransformer
>>> from blocks import BlockPipeline, custom_log_callback
>>> 
>>> pipe = BlockPipeline([
...   ("scaler", StandardScaler()),
...   ("regression", EstimatorTransformer(LinearRegression()))
... ],
...   record="scaler",
...   log_callback=custom_log_callback
... )

Logs

>>> pipe.fit(df, y)
# [custom_log_callback:78] - [scaler][StandardScaler()] shape=(1000, 10) time=0s

Records

>>> predicted = pipe.transform(df)
>>> pipe.name_record
# 'scaler'
>>> pipe.record
# array([[ ...

(back to top)

Contributing

Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.

If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement". Don't forget to give the project a star! Thanks again!

  1. Fork the Project
  2. Create your Feature Branch (git checkout -b feature/AmazingFeature)
  3. Commit your Changes (git commit -m 'Add some AmazingFeature')
  4. Push to the Branch (git push origin feature/AmazingFeature)
  5. Open a Pull Request

we also recommend to have a look at project-template.

project-template is a template project for scikit-learn compatible extensions. It aids development of estimators that can be used in scikit-learn pipelines and (hyper)parameter search, while facilitating testing (including some API compliance), documentation, open source development, packaging, and continuous integration.

Refer to the Official Documentation to modify the template for your own scikit-learn contribution.

(back to top)

License

Distributed under the BSD-3 License. See LICENSE.txt for more information.

(back to top)

About

Extra blocks for scikit-learn features.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages