Skip to content

Commit 2b4f069

Browse files
committed
Change train() to fit()
1 parent 5359664 commit 2b4f069

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

src/apps/genetic-nn.js

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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
});

src/apps/genetic.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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
});

src/apps/nn.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

src/apps/nnm.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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
}

0 commit comments

Comments
 (0)