.set width 500
SELECT
top 1 *
from mldb.ONNXClassification(
on (SELECT id, rev_text AS txt FROM amazon_reviews)
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(28)')
ModelOutputFieldsMap('logits')
ShowModelProperties('True')
) as td
;
*** Query completed. One row found. One column returned.
*** Total elapsed time was 1 second.
ModelProperties
----------------------------------------------------------------------------------------------------------
Input(s): {name: input_ids, tensor: INT64[batch_size, sequence_length]}, {name: attention_mask, tensor: INT64[batch_size, sequence_length]} Output(s): {name: logits, tensor: FLOAT32[batch_size, 2]}