Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/tommyod/KDEpy
Kernel Density Estimation in Python
https://github.com/tommyod/KDEpy
data-analysis exploratory-data-analysis kernel-density-estimation python python3 statistics
Last synced: about 1 month ago
JSON representation
Kernel Density Estimation in Python
- Host: GitHub
- URL: https://github.com/tommyod/KDEpy
- Owner: tommyod
- License: bsd-3-clause
- Created: 2018-01-19T15:27:57.000Z (almost 7 years ago)
- Default Branch: master
- Last Pushed: 2024-10-29T07:06:52.000Z (about 2 months ago)
- Last Synced: 2024-10-29T07:24:37.244Z (about 2 months ago)
- Topics: data-analysis, exploratory-data-analysis, kernel-density-estimation, python, python3, statistics
- Language: Jupyter Notebook
- Homepage: https://kdepy.readthedocs.io/en/latest/
- Size: 11.5 MB
- Stars: 580
- Watchers: 20
- Forks: 91
- Open Issues: 23
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
[](https://zenodo.org/badge/latestdoi/118142261)
[](https://github.com/tommyod/KDEpy/actions/workflows/test_master.yml)
[](http://kdepy.readthedocs.io/en/latest/?badge=latest)
[](https://badge.fury.io/py/KDEpy)
[](https://pepy.tech/project/kdepy)
---------
Want to cite KDEpy in your work? See the [bottom right part of this website](https://zenodo.org/record/2392268) for citation information.
# [KDEpy](https://kdepy.readthedocs.io/en/latest/)
## About
This Python 3.8+ package implements various kernel density estimators (KDE).
Three algorithms are implemented through the same API: [`NaiveKDE`](https://kdepy.readthedocs.io/en/latest/API.html#naivekde), [`TreeKDE`](https://kdepy.readthedocs.io/en/latest/API.html#treekde) and [`FFTKDE`](https://kdepy.readthedocs.io/en/latest/API.html#fftkde).
The class [`FFTKDE`](https://kdepy.readthedocs.io/en/latest/API.html#fftkde) outperforms other popular implementations, see the [comparison page](https://kdepy.readthedocs.io/en/latest/comparison.html).
**The code is stable and in widespread use by practitioners and in other packages.**
*The code generating the above graph is found in [examples.py](https://github.com/tommyod/KDEpy/blob/master/docs/source/examples.py).*
## Installation
KDEpy is available through [PyPI](https://pypi.org/project/KDEpy/), and may be installed using `pip`:
```text
pip install KDEpy
```
If you have [trouble on Ubuntu](https://github.com/tommyod/KDEpy/issues/11), try running `sudo apt install libpython3.X-dev`, where `3.X` is your Python version.
## Example code and documentation
Below is an example showing an unweighted and weighted kernel density.
From the code below, it should be clear how to set the *kernel*, *bandwidth* (variance of the kernel) and *weights*.
See the [documentation](https://kdepy.readthedocs.io/en/latest/examples.html) for more examples.
```python
from KDEpy import FFTKDE
import matplotlib.pyplot as plt
customer_ages = [40, 56, 20, 35, 27, 24, 29, 37, 39, 46]
# Distribution of customers
x, y = FFTKDE(kernel="gaussian", bw="silverman").fit(customer_ages).evaluate()
plt.plot(x, y)
# Distribution of customer income (weight each customer by their income)
customer_income = [152, 64, 24, 140, 88, 64, 103, 148, 150, 132]
# The `bw` parameter can be manually set, e.g. `bw=5`
x, y = FFTKDE(bw="silverman").fit(customer_ages, weights=customer_income).evaluate()
plt.plot(x, y)
```
The package consists of three algorithms. Here's a brief explanation:
- [`NaiveKDE`](https://kdepy.readthedocs.io/en/latest/API.html#naivekde) - A naive computation. Supports d-dimensional data, variable bandwidth, weighted data and many kernel functions. Very slow on large data sets.
- [`TreeKDE`](https://kdepy.readthedocs.io/en/latest/API.html#treekde) - A tree-based computation. Supports the same features as the naive algorithm, but is faster at the expense of small inaccuracy when using a kernel without finite support. Good for evaluation on non-uniform, arbitrary grids.
- [`FFTKDE`](https://kdepy.readthedocs.io/en/latest/API.html#fftkde) - A very fast convolution-based computation. Supports weighted d-dimensional data and many kernels, but not variable bandwidth. Must be evaluated on an equidistant grid, the finer the grid the higher the accuracy. Data points may not be outside of the grid.
## Issues and contributing
### Issues
If you are having trouble using the package, please let me know by creating an [Issue on GitHub](https://github.com/tommyod/KDEpy/issues) and I'll get back to you.
### Contributing
Whatever your mathematical and Python background is, you are very welcome to contribute to KDEpy.
To contribute, fork the project, create a branch and submit and Pull Request.
Please follow these guidelines:
- Import as few external dependencies as possible.
- Use test driven development, have tests and docs for every method.
- Cite literature and implement recent methods.
- Unless it's a bottleneck computation, readability trumps speed.
- Employ object orientation, but resist the temptation to implement many methods -- stick to the basics.
- Follow PEP8.