Good flavor search in $SU(5)$: a machine learning approach