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('VARBYTE(8)')
ModelOutputFieldsMap('logits')
) as td
;
*** Query completed. 2 rows found. 2 columns returned.
*** Total elapsed time was 3 seconds.
id logits
-------------------- --------------------------------------------------------
183038 6108D8BF0A01E63F
1481005 02594040A71122C0