Skip to content

Commit a250c85

Browse files
authored
Fixing predictMaskedToken to work with first token masked (#36)
lgtm
1 parent 10fb90a commit a250c85

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

predictMaskedToken.m

+1-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242

4343
for i = 1:numel(pieces)
4444
tokens = fulltok.tokenize(pieces(i));
45-
if ~isempty(tokens)
45+
if ~isempty(tokens{1})
4646
% "" tokenizes to empty - awkward
4747
x = cat(2,x,fulltok.encode(tokens{1}));
4848
end

test/tpredictMaskedToken.m

+5-2
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
ValidText = iGetValidText;
1010
end
1111

12-
methods(Test)
12+
methods(Test)
1313
function verifyOutputDimSizes(test, Models, ValidText)
1414
inSize = size(ValidText);
1515
mdl = bert("Model", Models);
@@ -43,5 +43,8 @@ function inputWithoutMASKRemainsTheSame(test, Models)
4343
validText = struct('StringsAsColumns',manyStrs,...
4444
'StringsAsRows',manyStrs',...
4545
'ManyStrings',repmat(singleStr,3),...
46-
'SingleString',singleStr);
46+
'SingleString',singleStr,...
47+
'FirstWordMasked', "[MASK] the pace.",...
48+
'LastWordMasked', "Accelerating the [MASK].",...
49+
'SingleMaskToken', "[MASK]");
4750
end

0 commit comments

Comments
 (0)