Кодируйте столбцы A и B как векторы и подходите для модели классификации.Затем, после подгонки, вы можете делать прогнозы для различных входных данных A ((1, 0, 0) в качестве примера) и получать вероятности в векторе B ((0,25, 0,5, 0,1, ..., 0,15) в качестве примера),Таким образом, в этом случае значение 1 для A обслуживает значения (1, 2, 3, 10) с вероятностями выше.В зависимости от задачи вы можете выбрать какой-либо порог.
В зависимости от данных, которые вам нужно выбрать, метод кодирования (фиктивная или горячая), модель, выборка, метрика и т. Д.