code: @Override
      public void setWeights(Map<String, INDArray> weights) throws InvalidKerasConfigurationException {
          this.weights = new HashMap<>();
          // TODO: "embeddings" is incorrectly read as "s" for some applications
          if (weights.containsKey("s")) {
              INDArray kernel = weights.get("s");
              weights.remove("s");
              weights.put(conf.getLAYER_FIELD_EMBEDDING_WEIGHTS(), kernel);
          }
  
          if (!weights.containsKey(conf.getLAYER_FIELD_EMBEDDING_WEIGHTS()))
              throw new InvalidKerasConfigurationException(
                      "Parameter " + conf.getLAYER_FIELD_EMBEDDING_WEIGHTS() + " does not exist in weights");
          INDArray kernel = weights.get(conf.getLAYER_FIELD_EMBEDDING_WEIGHTS());
          if (this.zeroMasking) {
              kernel.putRow(0, Nd4j.zeros(kernel.columns()));
          }
          this.weights.put(DefaultParamInitializer.WEIGHT_KEY, kernel);
  
          if (weights.size() > 2) {
              Set<String> paramNames = weights.keySet();
              paramNames.remove(conf.getLAYER_FIELD_EMBEDDING_WEIGHTS());
              String unknownParamNames = paramNames.toString();
              log.warn("Attemping to set weights for unknown parameters: "
                      + unknownParamNames.substring(1, unknownParamNames.length() - 1));
          }
      }
docs: Set weights for layer.
  
  @param weights Embedding layer weights