Skip to content

Add standalone puppi, some fixes to the bucketing in HEPT, a few addition to the standalone script#475

Merged
jpata merged 10 commits intojpata:mainfrom
kawaho:hept_studies
Apr 8, 2026
Merged

Add standalone puppi, some fixes to the bucketing in HEPT, a few addition to the standalone script#475
jpata merged 10 commits intojpata:mainfrom
kawaho:hept_studies

Conversation

@kawaho
Copy link
Copy Markdown
Contributor

@kawaho kawaho commented Apr 2, 2026

  1. PUPPI
  • Added puppi.py, which is a standalone puppi script that runs on numpy arrays with no CMSSW/MLPF dependence
  • eval.py now runs puppi and save the results to parquet file for analysis
  1. HEPT
  • Increment hashed index along phi first, then eta so that tracks/clusters in opposite ends of eta won't be grouped together
  • Allow hash index to go across pi to -pi
  • Apply random shift to phi so that the buckets do not necessarily start at phi=-pi
  • Save the bucket index and coordinates to parquet files to understand how regions are grouped in HEPT
  1. General
  • Add evaluation only mode to just run evaluation only on saved model
  • Added a neutral PU FFN
  • Save the attention matrix to npz files for the ease of changing the plotting style

@jpata
Copy link
Copy Markdown
Owner

jpata commented Apr 7, 2026

Did you plan to commit the change of input features? That requires a change on the CMSSW side as well.
I think in general it makes sense to test new models from new features separately to avoid entangling effects.

@kawaho
Copy link
Copy Markdown
Contributor Author

kawaho commented Apr 7, 2026

I agree. I can do that in a separate PR. I should be finished with this one within today.

@kawaho kawaho changed the title Add standalone puppi and a few addition to the standalone script Add standalone puppi, some fixes to the bucketing in HEPT, a few addition to the standalone script Apr 7, 2026
@jpata jpata merged commit 616b3b1 into jpata:main Apr 8, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants