Redshift Classification of Optical Gamma-Ray Bursts using Supervised Learning
Modern satellite missions localize many gamma-ray bursts (GRBs), but spectroscopic redshifts remain scarce because rapid optical follow-up is often unavailable. Optical plateau classification differs fundamentally from its X-ray counterpart: heterogeneous ground-based follow-up, dust extinction, and chromatic afterglow behavior introduce greater incompleteness than in the X-ray data. We present a supervised machine-learning classification of long-duration GRBs (duration $>$ 2 s) with measured redshifts and possessing optical plateau emission. As predictors, we combine plateau and prompt-emission observables. Using 171 such GRBs, we evaluate binary classification --- whether a GRB is high- or low-$z$ --- at thresholds $z_t = 2.0, 2.5$, and $3.0$. The best classifier, at $z_t = 2.0$, achieves a training-set true positive rate (TPR) of 0.741 and an area under the curve (AUC) of 0.841, with a train-to-test AUC difference within 4 percent. The novelty of this work is a classifier built on an optical-plateau-selected sample. Prompt-emission and absorption variables are favored over plateau parameters at $z_t = 2.0$ on the raw (complete-case) data with the Least Absolute Shrinkage and Selection Operator (LASSO), thus allowing a reliable classifier without waiting for the plateau signatures in the light curve. The X-ray absorption column density, $\log(\mathrm{NH})$, and the prompt-emission peak flux, $\log(\mathrm{PeakFlux})$, are stably selected by both the optical and X-ray classifiers. We publicly release a web-based application for the optical classifier, complementing the available X-ray classifier \citep{dainotti2025grbredshiftclassifierfollowup}, to help the community rapidly select high-$z$ GRBs for follow-up.