@@ -12,24 +12,24 @@ using TextAnalysis
12
12
13
13
# train transformer
14
14
tfidf_transformer = MLJText. TfidfTransformer ()
15
- test = machine (tfidf_transformer, ngram_vec)
16
- MLJBase. fit! (test )
15
+ test_machine = machine (tfidf_transformer, ngram_vec)
16
+ MLJBase. fit! (test_machine )
17
17
18
18
# test
19
19
test_doc = ngrams (NGramDocument (" Another sentence ok" ))
20
- transform (test , [test_doc])
20
+ test1 = transform (test_machine , [test_doc])
21
21
@test sum (test1, dims= 2 )[1 ] == 0.0
22
22
@test size (test1) == (1 , 11 )
23
23
24
24
test_doc2 = ngrams (NGramDocument (" Listen Sam, today is not the day." ))
25
- transform (test , [test_doc2])
25
+ test2 = transform (test_machine , [test_doc2])
26
26
@test sum (test2, dims= 2 )[1 ] > 0.0
27
27
@test size (test2) == (1 , 11 )
28
28
29
29
test_doc3 = ngrams .(
30
30
Corpus (NGramDocument (" Another sentence ok" ), NGramDocument (" Listen Sam, today is not the day." ))
31
31
)
32
- transform (test , test_doc3)
32
+ test3 = transform (test_machine , test_doc3)
33
33
@test sum (test3, dims= 2 )[1 ] == 0.0
34
34
@test sum (test3, dims= 2 )[2 ] > 0.0
35
35
@test size (test3) == (2 , 11 )
0 commit comments