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)', 'ARGMAX')
ModelOutputFieldsMap('softmax(logits)=negative,positive', 'argmax(softmax(logits))=am')
) as td
;
*** Query completed. 2 rows found. 5 columns returned.
*** Total elapsed time was 2 seconds.
id negative positive am_index am_value
-------------------- ---------------------- ---------------------- ----------- ------------
681111 1.95246990188025E-004 9.99804794788361E-001 1 9.99804794788361E-001
1111911 9.99684572219849E-001 3.15414945362136E-004 0 9.99684572219849E-001