如何从 ML.Net 中的拥抱脸的预训练模型加载分词器

时间:2021-05-17 10:53:48

标签: c# deep-learning nlp ml.net huggingface-tokenizers

使用 ML.Net 中的 model.onnx 进行预测。

我有一个使用 PyTorch 构建的导出 onnx 模型。我需要将 input_ids、attention_mask 和其他输入提供给该模型以进行预测。

我在构建模型时使用了来自 Huggingface 的预训练模型的分词器,它对我的​​文本输入进行了分词。

现在,我想使用模型(下载的 onnx 模型)在 ML.Net 中进行预测,但我无法找到将输入数据构建到模型的方法,因为我不确定如何将所需的标记器加载到 ML.Net。

0 个答案:

没有答案