Method constructor

cart-pole/index.js:61–67

* Constructor of PolicyNetwork. * * @param {number | number[] | tf.LayersModel} hiddenLayerSizes * Can be any of the following * - Size of the hidden layer, as a single number (for a single hidden * layer) * - An Array of numbers (for any number of hidden layers). *

(hiddenLayerSizesOrModel)

   *   - An instance of tf.LayersModel.
   */
  constructor(hiddenLayerSizesOrModel) {
    if (hiddenLayerSizesOrModel instanceof tf.LayersModel) {
      this.policyNet = hiddenLayerSizesOrModel;
    } else {
      this.createPolicyNetwork(hiddenLayerSizesOrModel);
    }
  }

  /**
   * Create the underlying model of this policy network.