The studies of heavy flavor (charm or bottom) hadrons in relativistic collisions provide an undisputed testing ground for the theory of strong interactions, quantum chromodynamics (QCD). As the majority of the heavy flavor particles are produced in the initial stages of the heavy-ion collisions, they experience the whole QCD medium evolution. The lightest open charm meson, , and hidden charm vector meson, J/, are particularly useful as they are abundantly produced as compared to other open and hidden charm hadrons, respectively. The and J/ mesons that are either directly formed during initial scattering or as the decay products of higher charm stages are referred to as the prompt production, which is essential to probe the QCD medium. On the other hand, the non-prompt and J/ mesons are usually formed as the decay products of beauty hadrons and can provide a key understanding of beauty hadrons.
In this contribution, we use machine learning (ML) models to segregate the prompt and non-prompt productions of and J/ mesons in proton-proton (pp) collisions at TeV using the
track-level information of the particles like an experimental environment. We have used the PYTHIA8 event generator to simulate the events for the study, which provides a good qualitative description of experimental measurements. We have considered the and J/ decay channels for our study. To separate prompt from non-prompt sector of charmonia and open charm mesons, topological production of and J/ are considered. We have used XGBoost, CatBoost, and Random Forest models for related studies, whereas for J/, we have used XGBoost and LighGBM models. For , we have used invariant mass ( ), pseudoproper time ( ), pseudoproper decay length ( ), and distance of closest approach ( ) as the training inputs. For J/ meson, the input sample is chosen keeping ALICE Run 3 muon forward tracker upgrade in mind, which includes, invariant mass ( ), transverse momentum, pseudorapidity, and . The machine learning models provide up to 99\% accuracy to dissect the prompt and non-prompt production of both and J/. Transverse momentum, rapidity and multiplicity differential comparisons between the true and predicted values are compared to evaluate the performance of the models. Experimental comparisons are also made wherever applicable. The ML methods used in the present study can replace the traditionally used fitting method with the added advantage of track label identification. The present ML-based identification of prompt and non-prompt charm hadrons can be useful in experiments that require precise measurements.