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 }) {
4444 console . info ( 'Training' ) ;
4545 // Do the NN training first
4646 tRexes . forEach ( ( tRex ) => {
47- tRex . model . train ( tRex . training . inputs , tRex . training . labels ) ;
47+ tRex . model . fit ( tRex . training . inputs , tRex . training . labels ) ;
4848 } ) ;
4949 // Genetic training
5050 const chromosomes = rankList . map ( ( tRex ) => tRex . model . getChromosome ( ) ) ;
5151 // Clear rankList
5252 rankList . splice ( 0 ) ;
53- geneticModel . train ( chromosomes ) ;
53+ geneticModel . fit ( chromosomes ) ;
5454 tRexes . forEach ( ( tRex , i ) => {
5555 tRex . model . setChromosome ( chromosomes [ i ] ) ;
5656 } ) ;
Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ function handleReset({ tRexes }) {
4242 const chromosomes = rankList . map ( ( tRex ) => tRex . model . getChromosome ( ) ) ;
4343 // Clear rankList
4444 rankList . splice ( 0 ) ;
45- geneticModel . train ( chromosomes ) ;
45+ geneticModel . fit ( chromosomes ) ;
4646 tRexes . forEach ( ( tRex , i ) => {
4747 tRex . model . setChromosome ( chromosomes [ i ] ) ;
4848 } ) ;
Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ function handleReset({ tRexes }) {
3434 } else {
3535 // Train the model before restarting.
3636 console . info ( 'Training' ) ;
37- tRex . model . train ( tRex . training . inputs , tRex . training . labels ) ;
37+ tRex . model . fit ( tRex . training . inputs , tRex . training . labels ) ;
3838 }
3939}
4040
Original file line number Diff line number Diff line change @@ -40,7 +40,7 @@ function handleReset({ tRexes }) {
4040 // Train the model before restarting.
4141 console . info ( 'Training' ) ;
4242 tRexes . forEach ( ( tRex ) => {
43- tRex . model . train ( training . inputs , training . labels ) ;
43+ tRex . model . fit ( training . inputs , training . labels ) ;
4444 } ) ;
4545 }
4646}
You can’t perform that action at this time.
0 commit comments