/** * Constructor from parsed Keras layer configuration dictionary. * * @param layerConfig dictionary containing Keras layer configuration * @param enforceTrainingConfig whether to enforce training-related configuration options * @throws InvalidKerasConfigurationException * @throws UnsupportedKerasConfigurationException */ public KerasGlobalPooling(Map<String, Object> layerConfig, boolean enforceTrainingConfig) throws InvalidKerasConfigurationException, UnsupportedKerasConfigurationException { super(layerConfig, enforceTrainingConfig); this.dimensions = mapPoolingDimensions(this.className); GlobalPoolingLayer.Builder builder = new GlobalPoolingLayer.Builder(mapPoolingType(this.className)).poolingDimensions(dimensions) .collapseDimensions(true).name(this.layerName).dropOut(this.dropout); this.layer = builder.build(); this.vertex = null; }