search

francois-drielsma / matcha

Repository housing python modules to aid in matching CRT hits to TPC tracks.
MIT License
1 stars 0 forks source link

Compare PCA to `centroid` direction estimation #12

Open francois-drielsma opened 1 year ago

francois-drielsma commented 1 year ago

I would recommend testing the centroid method to estimating the track direction. Here's the lartpc_mrleco3d implementation:

https://github.com/francois-drielsma/lartpc_mlreco3d/blob/d720174b9fdd23e616d3f15c6772a1a720b2b4e8/mlreco/utils/gnn/cluster.py#L564-L628

francois-drielsma commented 1 year ago

Here is the local np.mean implementation:

https://github.com/francois-drielsma/lartpc_mlreco3d/blob/d720174b9fdd23e616d3f15c6772a1a720b2b4e8/mlreco/utils/numba_local.py#L69-L95

and the local scipy.spatial.distance.cdist implementation:

https://github.com/francois-drielsma/lartpc_mlreco3d/blob/d720174b9fdd23e616d3f15c6772a1a720b2b4e8/mlreco/utils/numba_local.py#L318-L340

andrewmogan commented 1 year ago

Note to self: update README with this information when/if implemented