File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -9,7 +9,7 @@ const RNN_MODEL_URL = 'https://raw.githubusercontent.com/ml5js/ml5-data-and-mode
9
9
10
10
const RNN_MODEL_DEFAULTS = {
11
11
cellsAmount : 2 ,
12
- vocabSize : 223
12
+ vocabSize : 64
13
13
} ;
14
14
15
15
const RNN_DEFAULTS = {
@@ -21,15 +21,15 @@ const RNN_DEFAULTS = {
21
21
22
22
const RNN_OPTIONS = {
23
23
seed : 'the meaning of pizza is: ' ,
24
- length : 500 ,
24
+ length : 100 ,
25
25
temperature : 0.7
26
26
}
27
27
28
28
describe ( 'charRnn' , ( ) => {
29
29
let rnn ;
30
30
31
31
beforeAll ( async ( ) => {
32
- jasmine . DEFAULT_TIMEOUT_INTERVAL = 60000 ; //set extra long interval due to issues with CharRNN generation time
32
+ jasmine . DEFAULT_TIMEOUT_INTERVAL = 120000 ; //set extra long interval due to issues with CharRNN generation time
33
33
rnn = await charRNN ( RNN_MODEL_URL , undefined ) ;
34
34
} ) ;
35
35
@@ -54,7 +54,7 @@ describe('charRnn', () => {
54
54
55
55
it ( 'generates content that follows the set options' , async ( ) => {
56
56
const result = await rnn . generate ( RNN_OPTIONS ) ;
57
- expect ( result . sample . length ) . toBe ( 500 ) ;
57
+ expect ( result . sample . length ) . toBe ( 100 ) ;
58
58
} ) ;
59
59
} ) ;
60
60
} ) ;
You can’t perform that action at this time.
0 commit comments