Mean shift kernel update functions

kernel_update_functions.py

import numpy as np
from sklearn.metrics.pairwise import euclidean_distances

def gaussian_kernel_update(x, points, bandwidth):
    distances = euclidean_distances(points, x)
    weights = np.exp(-1 * (distances ** 2 / bandwidth ** 2))
    return np.sum(points * weights, axis=0) / np.sum(weights)

def flat_kernel_update(x, points, bandwidth):
    return np.mean(points, axis=0)