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('logits=negative,positive', 'argmax(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
-------------------- ---------------------- ---------------------- ----------- ----------------------
1353637 1.23476803302765E 000 -9.70530867576599E-001 0 1.23476803302765E 000
1556394 -2.39546513557434E 000 2.44996571540833E 000 1 2.44996571540833E 000