Skip to content

Nim implementation of a KDE function using Deriche approximation and linear binning.

License

Notifications You must be signed in to change notification settings

chancyk/deriche-kde

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

About

This code is a port of a port of a port, but it seems to work. If anyone that actually understands the math wants to takeover this codebase, please let me know.

The Deriche filter is an order-K approximation of the Gaussian kernel, which is combined with linear binning to produce a fast and accurate implementation of KDE. The naive KDE algorithm is O(n^2), while this implementation is O(n+m) wherem is the number of bins.

Usage

import derichekde

var data = @[1.2, 2.3, 23, 40, 50, 60, 70, 60, 50, 400, 700, 1000]
let (densities, lo, hi) = density_1d(data)

# The default bin size is 512, pass bins to change.
doAssert densities.len == 512

lo and hi are the minimum and maximum values in the data with a padding if extent is not provided.

Pass bandwidth to change the smoothness of the curve.

References

This code is a port of this Python implementation of the Deriche approximation KDE algorithm found here: https://github.com/liuzh-buaa/fast_kde

Which is itself a port of Javascript implementation... which was the implementation of a paper that was based on a port of the Deriche computer vision kernel implemented in C by Getreuer.

The original Javascript implementation is here: https://github.com/uwdata/fast-kde

The paper on the implementation for KDE can be found here: https://idl.uw.edu/papers/fast-kde

About

Nim implementation of a KDE function using Deriche approximation and linear binning.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages