Skip to main content

Package to build predictive models using linear/logistic regression

Project description

cobra is a Python package to build predictive models using linear/logistic regression with a focus on performance and interpretation. It consists of several modules for data preprocessing, feature selection and model evaluation. The underlying methodology was developed at Python Predictions in the course of hundreds of business-related prediction challenges. It has been tweaked, tested and optimized over the years based on feedback from clients, our team, and academic researchers.

Main Features

  • Prepare a given pandas DataFrame for predictive modelling: - partition into train/selection/validation sets - create bins from continuous variables - regroup categorical variables based on statistical significance - replace missing values and - add columns with incidence rate per category/bin.

  • Perform univariate feature selection based on AUC

  • Compute correlation matrix of predictors

  • Find the suitable variables using forward feature selection

  • Evaluate model performance and visualize the results

Getting started

These instructions will get you a copy of the project up and running on your local machine for usage, development and testing purposes.

Requirements

This package requires the usual Python packages for data science:

  • numpy (>=1.19.4)

  • pandas (>=1.1.5)

  • scipy (>=1.5.4)

  • scikit-learn (>=0.23.1)

  • matplotlib (>=3.3.3)

  • seaborn (>=0.11.0)

These packages, along with their versions are listed in requirements.txt and can be installed using pip:

pip install -r requirements.txt

Note: if you want to install cobra with e.g. pip, you don’t have to install all of these requirements as these are automatically installed with cobra itself.

Installation

The easiest way to install cobra is using pip

pip install -U pythonpredictions-cobra

Contributing to cobra

We’d love you to contribute to the development of cobra! There are many ways in which you can contribute, the most common of which is to contribute to the source code or documentation of the project. However, there are many other ways you can contribute (report issues, improve code coverage by adding unit tests, …). We use GitHub issue to track all bugs and feature requests. Feel free to open an issue in case you found a bug or in case you wish to see a new feature added.

How to contribute code

The preferred way to contribute to cobra is to fork the main repository on GitHub, then submit a “pull request” (PR). The first step is to get a local development copy by installing cobra from source through the following steps:

  • Fork the project repository. For more details on how to fork a repository see this guide

  • Clone your fork of cobra’s repo.

  • Open a shell and navigate to the folder where this repo was cloned in.

  • Once you are in the folder, execute pip install --editable ..

  • Create a feature branch to do your development.

  • Once your are finished developing, you can create a pull request from your fork (see this guide for detailed instructions).

Notes

  • Make sure to follow the PEP 8 styleguide if you make any changes to cobra. You should also write or modify unit test for your changes.

  • To avoid duplicating work, it is highly recommended that you search through the issue tracker and/or the PR list. If in doubt, you can always reach out to us through email (cobra@pythonpredictions.com)

Help and Support

Documentation

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

pythonpredictions-cobra-1.0.1.tar.gz (30.3 kB view hashes)

Uploaded Source

Built Distribution

pythonpredictions_cobra-1.0.1-py3-none-any.whl (37.0 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page