17 references to Pieces
Microsoft.ML.Tokenizers (17)
Model\SentencePieceBpeModel.cs (2)
28for (int i = 0; i < modelProto.Pieces.Count; i++) 30var piece = modelProto.Pieces[i];
Model\SentencePieceUnigramModel.cs (15)
31if (modelProto.TrainerSpec.BosId >= modelProto.Pieces.Count || 32modelProto.TrainerSpec.EosId >= modelProto.Pieces.Count || 33modelProto.TrainerSpec.UnkId >= modelProto.Pieces.Count) 38_vocabReverse = new (string Piece, float Score, ModelProto.Types.SentencePiece.Types.Type Type)[modelProto.Pieces.Count]; 43for (int i = 0; i < modelProto.Pieces.Count; i++) 45if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Normal || 46modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.UserDefined || 47modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Unused) 49string piece = modelProto.Pieces[i].Piece; 50float score = modelProto.Pieces[i].Score; 51_vocabReverse[i] = (piece, score, modelProto.Pieces[i].Type); 56else if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Byte) 60else if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Unknown) 63_vocabReverse[i] = (modelProto.Pieces[i].Piece, modelProto.Pieces[i].Score, ModelProto.Types.SentencePiece.Types.Type.Unknown);