HEP ML LAB: An end-to-end framework for applying machine learning to phenomenology studies

Recent years have seen the development and growth of machine learning in high-energy physics. However, additional effort is required to continue exploring the use of machine learning to its full potential. To simplify the application of the existing algorithms and neural networks and to advance the reproducibility of the analysis, we developed HEP ML LAB ( ), a Python-based, end-to-end framework for phenomenology studies. It covers the complete workflow from event generation to performance evaluation, and provides a consistent style of use for different approaches. We propose an observable naming convention to streamline the data extraction and conversion processes. In the KERAS style, we provide the traditional cut-and-count and boosted decision trees together with neural networks. We take the tagging as an example and evaluate all built-in approaches with the metrics of significance and background rejection. With its modular design, HEP ML LAB is easy to extend and customize, and can be used as a tool for both beginners and experienced researchers.