The macro will be used for the invariant mass calculation of omega via their di-muon decay channel with the help of machine learning classifiers from the TMVA class of ROOT. The improvement in reconstructed efficiency and S/B ratio are improved with the help of ML. Macro should be added into the macro/analysis/much/