python - 如何使用 scikit-learn knn 将方向数据用作特征？

Question

我是使用 KNN 的新手，在我的火车组中我有速度矢量。由于方向 359° 和 0° 完全不同，我正在考虑转换方向，以便测试数据中的矢量指向 180°。

如果我从一个数据点进行预测，我可以在使用 KNeighborsClassifier 之前进行这种转换，但是当我使用 GridSearchCV 调整超参数时，应该在每次比较之间进行转换。有没有办法做到这一点？

或者我想念的其他方式？

Answer 1

您可以将方向列theta分成2 列sin(theta)和cos(theta)，这两个列都具有连续性。