Reactor-based models are well-suited Turbulence-Chemistry Interactions (TCI) Sub-Grid Scale (SGS) closures for Large Eddy Simulation (LES) due to their ability to account for finite-rate kinetics. The Partially Stirred Reactor (PaSR) model relies on the estimation of characteristic time scales to define the reacting fraction of each computational cell. However, chemistry develops a spectrum of intrinsic chemical time scales, leaving no clear consensus on the definition of a single representative scale. Nevertheless, in numerical codes, a single chemical time scale formulation is used on the whole physical domain despite local and complex phenomena. Through an a priori assessment on Direct Numerical Simulation (DNS) data of turbulent non-premixed combustion, the present work proposes a numerical method to locally select an optimal chemical time scale formulation that minimises the model error. Data points are grouped into clusters via supervised partitioning algorithms where the optimal formulation is attributed to each cluster by means of distances minimisation. Using a combination of partitioning procedures can further improve the reconstruction quality of the clustered solutions, up to 35% global errors reductions with respect to standard solutions. Existing data partitions are then tested on unseen data points, yielding great prediction capabilities across grid