diff --git a/lstm/lstm.js b/lstm/lstm.js index fdf2e4c..6795532 100644 --- a/lstm/lstm.js +++ b/lstm/lstm.js @@ -70,7 +70,7 @@ export class LSTM { verbose: 0, callbacks: { onEpochEnd: (epoch, logs) => { - const percent = Math.round(((epoch + 1) / 30) * 100); + const percent = Math.round(((epoch + 1) / this.settings.epochs) * 100); console.log(`Training... ${percent}% (Epoch ${epoch + 1}/${this.settings.epochs}, loss=${logs.loss.toFixed(4)})`); } } @@ -194,4 +194,4 @@ export class LSTM { } return generated; } -} \ No newline at end of file +}