FLATTEN результаты с использованием значения MAX в BigQuery

Question

Мне нужно сгладить столбец вероятностей в моих результатах с максимальным числом:

original predicted probabilities
 <=50K   >50K      >50K 0.5377828170971353
                   <=50K 0.46221718290286473
 <=50K   <=50K     >50K 0.05434716579642335
                   <=50K 0.9456528342035766

Я бы хотел сгладить свой результат, но теперь, используя этот запрос, я просто получаю таблицу выше и использую Python-клиент bigQueryполучить: [object Object],[object Object]

    SELECT
      original,
      predicted,
      probabilities
    FROM
      ML.PREDICT(MODEL `my_dataset.my_model`,
      (
      SELECT
        *
      FROM   
       `bigquery-public-data.ml_datasets.census_adult_income`

      ))