Method createModel

lstm-text-generation/index.js:63–66

* Create LSTM model from scratch. * * @param {number | number[]} lstmLayerSizes Sizes of the LSTM layers, as a * number or an non-empty array of numbers.

(lstmLayerSizes)

   *   number or an non-empty array of numbers.
   */
  createModel(lstmLayerSizes) {
    this.model = model.createModel(
        this.sampleLen_, this.charSetSize_, lstmLayerSizes);
  }

  /**
   * Compile model for training.