add layerClass attribute to layers

src/Layer.js

@@ -10,6 +10,7 @@ export default class Layer {
    * @param {Object} [attrs] - layer attributes
    */
   constructor (attrs = {}) {
+    this.layerClass = 'Layer'
     this.name = attrs.name
 
     this.params = []

src/layers/advanced_activations/ELU.js

@@ -11,6 +11,7 @@ export default class ELU extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'ELU'
 
     const {
       alpha = 1.0

src/layers/advanced_activations/LeakyReLU.js

@@ -11,6 +11,7 @@ export default class LeakyReLU extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'LeakyReLU'
 
     const {
       alpha = 0.3

src/layers/advanced_activations/PReLU.js

@@ -16,6 +16,7 @@ export default class PReLU extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'PReLU'
 
     // Layer weights specification
     this.params = ['alphas']

src/layers/advanced_activations/ParametricSoftplus.js

@@ -11,6 +11,7 @@ export default class ParametricSoftplus extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'ParametricSoftplus'
 
     // Layer weights specification
     this.params = ['alphas', 'betas']

src/layers/advanced_activations/SReLU.js

@@ -11,6 +11,7 @@ export default class SReLU extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'SReLU'
 
     // Layer weights specification
     this.params = ['t_left', 'a_left', 't_right', 'a_right']

src/layers/advanced_activations/ThresholdedReLU.js

@@ -11,6 +11,7 @@ export default class ThresholdedReLU extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'ThresholdedReLU'
 
     const {
       theta = 1.0

src/layers/convolutional/AtrousConvolution2D.js

@@ -20,6 +20,8 @@ export default class AtrousConvolution2D extends Convolution2D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'AtrousConvolution2D'
+
     const {
       atrousRate = [1, 1]
     } = attrs

src/layers/convolutional/Convolution1D.js

@@ -15,6 +15,8 @@ export default class Convolution1D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Convolution1D'
+
     const {
       nbFilter = 1,
       filterLength = 1,

src/layers/convolutional/Convolution2D.js

@@ -19,6 +19,8 @@ export default class Convolution2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Convolution2D'
+
     const {
       nbFilter = 1,
       nbRow = 3,

src/layers/convolutional/Convolution3D.js

@@ -20,6 +20,8 @@ export default class Convolution3D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Convolution3D'
+
     const {
       nbFilter = 1,
       kernelDim1 = 1,

src/layers/convolutional/Deconvolution2D.js

@@ -21,6 +21,8 @@ export default class Deconvolution2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Deconvolution2D'
+
     const {
       nbFilter = 1,
       nbRow = 1,

src/layers/convolutional/SeparableConvolution2D.js

@@ -17,6 +17,8 @@ export default class SeparableConvolution2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'SeparableConvolution2D'
+
     const {
       nbFilter = 1,
       nbRow = 1,

src/layers/convolutional/UpSampling1D.js

@@ -12,6 +12,8 @@ export default class UpSampling1D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'UpSampling1D'
+
     const {
       length = 2
     } = attrs

src/layers/convolutional/UpSampling2D.js

@@ -12,6 +12,8 @@ export default class UpSampling2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'UpSampling2D'
+
     const {
       size = [2, 2],
       dimOrdering = 'tf'

src/layers/convolutional/UpSampling3D.js

@@ -12,6 +12,8 @@ export default class UpSampling3D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'UpSampling3D'
+
     const {
       size = [2, 2, 2],
       dimOrdering = 'tf'

src/layers/convolutional/ZeroPadding1D.js

@@ -12,6 +12,8 @@ export default class ZeroPadding1D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'ZeroPadding1D'
+
     const {
       padding = 1
     } = attrs

src/layers/convolutional/ZeroPadding2D.js

@@ -12,6 +12,8 @@ export default class ZeroPadding2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'ZeroPadding2D'
+
     const {
       padding = [1, 1],
       dimOrdering = 'tf'

src/layers/convolutional/ZeroPadding3D.js

@@ -12,6 +12,8 @@ export default class ZeroPadding3D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'ZeroPadding3D'
+
     const {
       padding = [1, 1, 1],
       dimOrdering = 'tf'

src/layers/core/Activation.js

@@ -11,6 +11,7 @@ export default class Activation extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Activation'
 
     const {
       activation = 'linear'

src/layers/core/Dense.js

@@ -15,6 +15,8 @@ export default class Dense extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Dense'
+
     const {
       outputDim = 1,
       activation = 'linear',

src/layers/core/Dropout.js

@@ -11,6 +11,7 @@ export default class Dropout extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Dropout'
 
     const {
       p = 0.5

src/layers/core/Flatten.js

@@ -14,6 +14,7 @@ export default class Flatten extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Flatten'
   }
 
   /**

src/layers/core/Highway.js

@@ -17,6 +17,8 @@ export default class Highway extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Highway'
+
     const {
       transformBias = -2,
       activation = 'linear',

src/layers/core/MaxoutDense.js

@@ -18,6 +18,8 @@ export default class MaxoutDense extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'MaxoutDense'
+
     const {
       outputDim = 1,
       inputDim = null,

src/layers/core/Merge.js

@@ -17,6 +17,8 @@ export default class Merge extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Merge'
+
     const {
       mode = 'sum',
       concatAxis = -1,

src/layers/core/Permute.js

@@ -12,6 +12,8 @@ export default class Permute extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Permute'
+
     const {
       dims = []
     } = attrs

src/layers/core/RepeatVector.js

@@ -14,6 +14,8 @@ export default class RepeatVector extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'RepeatVector'
+
     const {
       n = 1
     } = attrs

src/layers/core/Reshape.js

@@ -15,6 +15,8 @@ export default class Reshape extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Reshape'
+
     const {
       shape = []
     } = attrs

src/layers/embeddings/Embedding.js

@@ -11,6 +11,8 @@ export default class Embedding extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'Embedding'
+
     const {
       inputDim = 1,
       outputDim = 1,

src/layers/normalization/BatchNormalization.js

@@ -13,6 +13,8 @@ export default class BatchNormalization extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'BatchNormalization'
+
     const {
       epsilon = 1e-5,
       mode = 0,

src/layers/pooling/AveragePooling1D.js

@@ -9,6 +9,8 @@ export default class AveragePooling1D extends _Pooling1D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'AveragePooling1D'
+
     this.poolingFunc = 'average'
   }
 }

src/layers/pooling/AveragePooling2D.js

@@ -9,6 +9,8 @@ export default class AveragePooling2D extends _Pooling2D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'AveragePooling2D'
+
     this.poolingFunc = 'average'
   }
 }

src/layers/pooling/AveragePooling3D.js

@@ -9,6 +9,8 @@ export default class AveragePooling3D extends _Pooling3D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'AveragePooling3D'
+
     this.poolingFunc = 'average'
   }
 }

src/layers/pooling/GlobalAveragePooling1D.js

@@ -11,6 +11,7 @@ export default class GlobalAveragePooling1D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'GlobalAveragePooling1D'
   }
 
   /**

src/layers/pooling/GlobalAveragePooling2D.js

@@ -11,6 +11,8 @@ export default class GlobalAveragePooling2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'GlobalAveragePooling2D'
+
     const {
       dimOrdering = 'tf'
     } = attrs

src/layers/pooling/GlobalMaxPooling1D.js

@@ -11,6 +11,7 @@ export default class GlobalMaxPooling1D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'GlobalMaxPooling1D'
   }
 
   /**

src/layers/pooling/GlobalMaxPooling2D.js

@@ -11,6 +11,8 @@ export default class GlobalMaxPooling2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'GlobalMaxPooling2D'
+
     const {
       dimOrdering = 'tf'
     } = attrs

src/layers/pooling/MaxPooling1D.js

@@ -9,6 +9,8 @@ export default class MaxPooling1D extends _Pooling1D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'MaxPooling1D'
+
     this.poolingFunc = 'max'
   }
 }

src/layers/pooling/MaxPooling2D.js

@@ -9,6 +9,8 @@ export default class MaxPooling2D extends _Pooling2D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'MaxPooling2D'
+
     this.poolingFunc = 'max'
   }
 }

src/layers/pooling/MaxPooling3D.js

@@ -9,6 +9,8 @@ export default class MaxPooling3D extends _Pooling3D {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = 'MaxPooling3D'
+
     this.poolingFunc = 'max'
   }
 }

src/layers/pooling/_Pooling1D.js

@@ -11,6 +11,8 @@ export default class _Pooling1D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = '_Pooling1D'
+
     const {
       poolLength = 2,
       stride = null,

src/layers/pooling/_Pooling2D.js

@@ -11,6 +11,8 @@ export default class _Pooling2D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = '_Pooling2D'
+
     const {
       poolSize = [2, 2],
       strides = null,

src/layers/pooling/_Pooling3D.js

@@ -11,6 +11,8 @@ export default class _Pooling3D extends Layer {
    */
   constructor (attrs = {}) {
     super(attrs)
+    this.layerClass = '_Pooling3D'
+
     const {
       poolSize = [2, 2, 2],
       strides = null,