File tree Expand file tree Collapse file tree 4 files changed +5
-5
lines changed Expand file tree Collapse file tree 4 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -44,13 +44,13 @@ function handleReset({ tRexes }) {
44
44
console . info ( 'Training' ) ;
45
45
// Do the NN training first
46
46
tRexes . forEach ( ( tRex ) => {
47
- tRex . model . train ( tRex . training . inputs , tRex . training . labels ) ;
47
+ tRex . model . fit ( tRex . training . inputs , tRex . training . labels ) ;
48
48
} ) ;
49
49
// Genetic training
50
50
const chromosomes = rankList . map ( ( tRex ) => tRex . model . getChromosome ( ) ) ;
51
51
// Clear rankList
52
52
rankList . splice ( 0 ) ;
53
- geneticModel . train ( chromosomes ) ;
53
+ geneticModel . fit ( chromosomes ) ;
54
54
tRexes . forEach ( ( tRex , i ) => {
55
55
tRex . model . setChromosome ( chromosomes [ i ] ) ;
56
56
} ) ;
Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ function handleReset({ tRexes }) {
42
42
const chromosomes = rankList . map ( ( tRex ) => tRex . model . getChromosome ( ) ) ;
43
43
// Clear rankList
44
44
rankList . splice ( 0 ) ;
45
- geneticModel . train ( chromosomes ) ;
45
+ geneticModel . fit ( chromosomes ) ;
46
46
tRexes . forEach ( ( tRex , i ) => {
47
47
tRex . model . setChromosome ( chromosomes [ i ] ) ;
48
48
} ) ;
Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ function handleReset({ tRexes }) {
34
34
} else {
35
35
// Train the model before restarting.
36
36
console . info ( 'Training' ) ;
37
- tRex . model . train ( tRex . training . inputs , tRex . training . labels ) ;
37
+ tRex . model . fit ( tRex . training . inputs , tRex . training . labels ) ;
38
38
}
39
39
}
40
40
Original file line number Diff line number Diff line change @@ -40,7 +40,7 @@ function handleReset({ tRexes }) {
40
40
// Train the model before restarting.
41
41
console . info ( 'Training' ) ;
42
42
tRexes . forEach ( ( tRex ) => {
43
- tRex . model . train ( training . inputs , training . labels ) ;
43
+ tRex . model . fit ( training . inputs , training . labels ) ;
44
44
} ) ;
45
45
}
46
46
}
You can’t perform that action at this time.
0 commit comments