ONNXClassification Example: Logits as FLOAT32(2) – Softmax Output - Teradata Vantage

SELECT
    *
from mldb.ONNXClassification(
    on (SELECT id, rev_text AS txt FROM amazon_reviews sample 2)
    on (SELECT model_id, model FROM classification_models WHERE model_id = 'distilbert-base-uncased-finetuned-sst-2-english') DIMENSION
    on (SELECT tokenizer FROM classification_tokenizers WHERE model_id = 'distilbert-base-uncased-finetuned-sst-2-english') DIMENSION
USING
    Accumulate('id')
    OutputFormat('FLOAT32(2)')
    ModelOutputFieldsMap('softmax(logits)=negative,positive')
) as td
;
 
*** Query completed. 2 rows found. 3 columns returned.
*** Total elapsed time was 2 seconds.
 

id                   negative               positive
-------------------- ---------------------- ----------------------
1353637              9.00724351406097E-001  9.92756560444832E-002
382777               9.99751508235931E-001  2.48541502514854E-004