Skip to content

Commit 09ad027

Browse files
committed
Reducing set of models for testing.
1 parent 19c0a1e commit 09ad027

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

Diff for: test/tpredictMaskedToken.m

+7-9
Original file line numberDiff line numberDiff line change
@@ -5,30 +5,28 @@
55
% Copyright 2023 The MathWorks, Inc.
66

77
properties(TestParameter)
8-
AllModels = {"base","multilingual-cased","medium",...
9-
"small","mini","tiny","japanese-base",...
10-
"japanese-base-wwm"}
8+
Models = {"tiny","japanese-base-wwm"}
119
ValidText = iGetValidText;
1210
end
1311

1412
methods(Test)
15-
function verifyOutputDimSizes(test, AllModels, ValidText)
13+
function verifyOutputDimSizes(test, Models, ValidText)
1614
inSize = size(ValidText);
17-
mdl = bert("Model", AllModels);
15+
mdl = bert("Model", Models);
1816
outputText = predictMaskedToken(mdl,ValidText);
1917
test.verifyEqual(size(outputText), inSize);
2018
end
2119

22-
function maskTokenIsRemoved(test, AllModels)
20+
function maskTokenIsRemoved(test, Models)
2321
text = "This has a [MASK] token.";
24-
mdl = bert("Model", AllModels);
22+
mdl = bert("Model", Models);
2523
outputText = predictMaskedToken(mdl,text);
2624
test.verifyFalse(contains(outputText, "[MASK]"));
2725
end
2826

29-
function inputWithoutMASKRemainsTheSame(test, AllModels)
27+
function inputWithoutMASKRemainsTheSame(test, Models)
3028
text = "This has a no mask token.";
31-
mdl = bert("Model", AllModels);
29+
mdl = bert("Model", Models);
3230
outputText = predictMaskedToken(mdl,text);
3331
test.verifyEqual(text, outputText);
3432
end

0 commit comments

Comments
 (0)