geo-features-generator 0.1.5

Sklearn-style generator of geo features: distances, angles, cosine similarity, KMeans, NN, polar features

geo-features-generator

Генератор гео-признаков в стиле sklearn: расстояния (Haversine, equirectangular, Manhattan), углы/синусы/косинусы, средние точки, дельты, квадраты расстояний, косинусная близость (3D), кластеризация KMeans, расстояния до ближайшего соседа, полярные признаки относительно базовой точки, H3 ячейки Uber.

Установка

pip install geo-features-generator

# Для H3 функций (опционально)
pip install geo-features-generator[h3]

Быстрый старт

import pandas as pd
from geo_features_generator import GeoFeaturesGenerator

df = pd.DataFrame({
    "lat1": [55.75, 59.93],
    "lon1": [37.62, 30.33],
    "lat2": [59.93, 55.75],
    "lon2": [30.33, 37.62],
})

gen = GeoFeaturesGenerator(
    coordinate_pairs=[("lat1", "lon1"), ("lat2", "lon2")],
    enable_squared_distances=True,
    enable_normed_deltas=True,
    enable_cosine_similarity=True,
    enable_kmeans=False,
    enable_nearest_neighbor=False,
    enable_polar_features=True,
    polar_base_point=(55.75, 37.62),
    enable_h3_features=True,
    h3_resolutions=[8, 10],
)

features = gen.fit_transform(df)
print(features.head())

Параметры

• coordinate_pairs: список пар имен столбцов (lat, lon).
• radius: радиус сферы (метры), по умолчанию 6_371_000.
• output_format: "pandas" или "numpy".
• generate_point_features: генерировать признаки для каждой точки.
• generate_pair_features: генерировать признаки для пар точек.
• enable_squared_distances: добавляет _m2 признаки квадратов расстояний.
• enable_normed_deltas: добавляет abs_dlat_m, abs_dlon_m (в метрах).
• enable_cosine_similarity: cosine_sim, cosine_dist по 3D dot на сфере.
• enable_kmeans: добавляет *_kmeans_label для каждой точки (требуется scikit-learn).
• kmeans_n_clusters, kmeans_random_state: параметры KMeans.
• enable_nearest_neighbor: добавляет *_nn_haversine_m (требуется scikit-learn).
• enable_polar_features: добавляет *_polar_bearing_deg, *_polar_dist_m относительно polar_base_point.
• polar_base_point: (lat, lon) базовой точки.
• enable_h3_features: добавляет H3 признаки (требует установленной библиотеки 'h3').
• h3_resolutions: разрешения H3 ячеек для генерации признаков, по умолчанию [6, 8, 10].
• h3_neighbor_rings: количество колец соседей для H3 ячеек, по умолчанию 1.
• h3_parent_resolutions: разрешения родительских H3 ячеек, по умолчанию [4, 6].
• h3_child_resolutions: разрешения дочерних H3 ячеек, по умолчанию [10, 12].

Контакты

• 
• 

Лицензия

MIT. См. файл LICENSE.