使用中位数（而不是均值）使用 sklearn-KNN 进行预测

Question

Sklearn-KNN 允许在计算 mean x 最近邻时设置权重（例如，均匀、距离）。

不是用均值预测，是否可以用中位数预测（也许用用户定义的函数）？

Answer 1

没有内置参数来调整权重以使用中位数而不是均值（您可以看到 in the source 均值是硬编码的）。但是因为 scikit-learn 估计器只是 Python 类，你可以子类化 KNeighborsRegressor 并覆盖 predict 方法来做任何你想做的事情。

这是一个简单的示例，我复制并粘贴了原始 predict() 方法并修改了相关部分：

from sklearn.neighbors.regression import KNeighborsRegressor, check_array, _get_weights

class MedianKNNRegressor(KNeighborsRegressor):
    def predict(self, X):
        X = check_array(X, accept_sparse='csr')

        neigh_dist, neigh_ind = self.kneighbors(X)

        weights = _get_weights(neigh_dist, self.weights)

        _y = self._y
        if _y.ndim == 1:
            _y = _y.reshape((-1, 1))

        ######## Begin modification
        if weights is None:
            y_pred = np.median(_y[neigh_ind], axis=1)
        else:
            # y_pred = weighted_median(_y[neigh_ind], weights, axis=1)
            raise NotImplementedError("weighted median")
        ######### End modification

        if self._y.ndim == 1:
            y_pred = y_pred.ravel()

        return y_pred    

X = np.random.rand(100, 1)
y = 20 * X.ravel() + np.random.rand(100)
clf = MedianKNNRegressor().fit(X, y)
print(clf.predict(X[:5]))
# [  2.38172861  13.3871126    9.6737255    2.77561858  17.07392584]

我省略了加权版本，因为我不知道使用 numpy/scipy 计算加权中位数的简单方法，但是一旦该函数可用，添加它会很简单。

使用中位数（而不是均值）使用 sklearn-KNN 进行预测

Predict with sklearn-KNN using median (instead of mean)

python

knn

scikit-learn

使用中位数（而不是均值）使用 sklearn-KN​​N 进行预测

Predict with sklearn-KNN using median (instead of mean)

python

knn

scikit-learn

使用中位数（而不是均值）使用 sklearn-KNN 进行预测