Metadata-Version: 2.1
Name: shml
Version: 0.1
Summary: lightweight pdata cleaning/processing/plotting/ML training library for use with an ATLAS BSM dihiggs search
Home-page: https://github.com/phinate/shml
Author: Nathan Simpson
Author-email: n.s@cern.ch
Maintainer: Nathan Simpson
Maintainer-email: n.s@cern.ch
License: BSD-3-Clause
Project-URL: Documentation, https://shml.readthedocs.io/
Project-URL: Bug Tracker, https://github.com/phinate/shml/issues
Project-URL: Discussions, https://github.com/phinate/shml/discussions
Project-URL: Changelog, https://github.com/phinate/shml/releases
Platform: Any
Classifier: Development Status :: 1 - Planning
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: BSD License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Topic :: Scientific/Engineering
Requires-Python: >=3.7
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: awkward (>=1.5.1)
Requires-Dist: pre-commit
Requires-Dist: pyarrow (>=6.0.0)
Requires-Dist: uproot (>=4.1.8)
Requires-Dist: typing-extensions (>=3.7) ; python_version < "3.8"
Provides-Extra: dev
Requires-Dist: pytest (>=6) ; extra == 'dev'
Provides-Extra: docs
Requires-Dist: Sphinx (~=3.0) ; extra == 'docs'
Requires-Dist: myst-parser (>=0.13) ; extra == 'docs'
Requires-Dist: sphinx-book-theme (>=0.1.0) ; extra == 'docs'
Requires-Dist: sphinx-copybutton ; extra == 'docs'
Provides-Extra: ml
Requires-Dist: jax ; extra == 'ml'
Requires-Dist: jaxlib ; extra == 'ml'
Requires-Dist: matplotlib ; extra == 'ml'
Requires-Dist: pytorch-lightning (>=1.5) ; extra == 'ml'
Provides-Extra: test
Requires-Dist: pytest (>=6) ; extra == 'test'



[![Actions Status][actions-badge]][actions-link]
[![Code style: black][black-badge]][black-link]
[![PyPI version][pypi-version]][pypi-link]
[![PyPI platforms][pypi-platforms]][pypi-link]
[![GitHub Discussion][github-discussions-badge]][github-discussions-link]


# `shml`: routines to automate machine learning experiments for a X -> SH -> bbyy search

This module aims to provide a set of functions that, when composed, can run a pipeline capable of:
- going from `.root` files to `parquet` files via `uproot` and `awkward`
- constructing useful kinematic quantites for training
- applying a chosen or manual preselection
- configuring any additional processing, e.g. weight normalization, feature scaling
- access event data that's prepared for `pytorch` using `shml.torch_dataset.EventDataset`

still to do:
- infra to run ml experiments in a GPU or CPU environment via `pytorch-lightining`

## Usage
To see currently usable implemented functionality, check the [`examples`](examples) folder.

## Install
For preprocessing only:
```
python3 -m pip install shml
```
For ML extras (`pytorch`, plotting):
```
python3 -m pip install shml[ml]
```
[actions-badge]:            https://github.com/phinate/shml/workflows/CI/badge.svg
[actions-link]:             https://github.com/phinate/shml/actions
[black-badge]:              https://img.shields.io/badge/code%20style-black-000000.svg
[black-link]:               https://github.com/psf/black
[conda-badge]:              https://img.shields.io/conda/vn/conda-forge/shml
[conda-link]:               https://github.com/conda-forge/shml-feedstock
[github-discussions-badge]: https://img.shields.io/static/v1?label=Discussions&message=Ask&color=blue&logo=github
[github-discussions-link]:  https://github.com/phinate/shml/discussions
[gitter-badge]:             https://badges.gitter.im/https://github.com/phinate/shml/community.svg
[gitter-link]:              https://gitter.im/https://github.com/phinate/shml/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge
[pypi-link]:                https://pypi.org/project/shml/
[pypi-platforms]:           https://img.shields.io/pypi/pyversions/shml
[pypi-version]:             https://badge.fury.io/py/shml.svg
[rtd-badge]:                https://readthedocs.org/projects/shml/badge/?version=latest
[rtd-link]:                 https://shml.readthedocs.io/en/latest/?badge=latest
[sk-badge]:                 https://scikit-hep.org/assets/images/Scikit--HEP-Project-blue.svg


