返回模型
说明文档
将此模型移植到 C# 中使用
-
C# 使用示例:
using AI.DataPrepaire.DataLoader.NNWBlockLoader; using AI.DataStructs.Algebraic; using AI.ONNX.NLP.Bert; string path = "SbertDistilAIFr"; // 加载 Bert 模型 BertEmbedder embedder = BertEmbedder.FromPretrained(path); // 添加最后(线性)层 LinearLayerLoader linearLayer = LinearLayerLoader.LoadFromBinary(@$"{path}\1_Linear\model.aifr"); embedder.V2VBlocks.Add(linearLayer); // 句子向量化 Vector vects1 = embedder.ForwardSBert("Kittens love milk"); Vector vects2 = embedder.ForwardSBert("Visual Studio Code (VS Code) — текстовый редактор, разработанный Microsoft для Windows, Linux и macOS."); Vector vects3 = embedder.ForwardSBert("Visual Studio Code, also commonly referred to as VS Code,[12] is a source-code editor developed by Microsoft for Windows, Linux and macOS."); Vector vects1_rus = embedder.ForwardSBert("Котята любят молоко"); // 计算文本之间的相似度 Console.WriteLine(vects1.Cos(vects1_rus)); Console.WriteLine(vects1.Cos(vects2)); Console.WriteLine(vects2.Cos(vects3)); Console.WriteLine(vects1.Cos(vects3)); Console.ReadKey(); -
网络输出
0.8312274467817271
0.18402859300814606
0.9556528309990797
0.14738954582438052
FractalGPT/SbertDistilAIFr
作者 FractalGPT
feature-extraction
transformers
↓ 1
♥ 0
创建时间: 2024-01-15 09:40:51+00:00
更新时间: 2024-01-16 00:49:09+00:00
在 Hugging Face 上查看文件 (7)
.gitattributes
1_Linear/model.aifr
README.md
config.json
model.onnx
ONNX
tokenizer_config.json
vocab.txt