Prompt and non-prompt production of charm hadrons in proton-proton collisions at the Large Hadron Collider using machine learning

In this contribution, we use machine learning (ML) based models to separate the prompt and non-prompt production of heavy flavour hadrons, such as $D^0$ and J/$\psi$, in proton-proton collisions at LHC energies. For this purpose, we use PYTHIA~8 to generate events, which provides a good qualitative agreement with experimental measurements of charm hadron production. The input features for the ML models are experimentally measurable. The prediction accuracy of the ML models used in this study reaches up to 99\%. The ML models can be useful in providing precise track-level identification, which is not possible in experiments with traditional methods. The contribution also discusses future applications of the ML models to understand the production of prompt and non-prompt heavy quark hadrons.