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('logits')
) as td
;
*** Query completed. 2 rows found. 3 columns returned.
*** Total elapsed time was 2 seconds.
id logits_0 logits_1
-------------------- ---------------------- ----------------------
1866331 2.87072730064392E 000 -2.36050724983215E 000
2651604 -4.02496862411499E 000 4.28052234649658E 000