Add potential todo

ARTIST-Association · Jul 17, 2024 · e6a0fa7 · e6a0fa7
1 parent dc38197
commit e6a0fa7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/plots/05_nearest_neighbor_splitting.py b/plots/05_nearest_neighbor_splitting.py
@@ -49,6 +49,7 @@ def sort(self, data : pd.DataFrame) -> Tuple[pd.DataFrame, pd.DataFrame]:
             raise RuntimeError(f"Insufficient data points given. Expected at least {self._number_of_train_samples + self._number_of_validation_samples} but got {len(data)}!")
 
         # sort data by nearest neighbor distance
+        # TODO maybe use kd-tree for distance calculation
         distances = cdist(data[[mappings.AZIMUTH, mappings.ELEVATION, mappings.ID_INDEX]], data[[mappings.AZIMUTH, mappings.ELEVATION, mappings.ID_INDEX]], metric=self._metric)
         minimum_distances = np.sort(distances, axis=1)[:,0]
         sorted_indices = np.argsort(minimum_distances, axis=0)