add descriptions to layers from attributes

src/Layer.js

@@ -12,6 +12,7 @@ export default class Layer {
   constructor(attrs = {}) {
     this.layerClass = 'Layer'
     this.name = attrs.name
+    this.description = ''
     this.gpu = webgl2.isSupported && attrs.gpu
 
     this.params = []

src/layers/InputLayer.js

@@ -18,6 +18,8 @@ export default class InputLayer extends Layer {
     const { shape = [] } = attrs
 
     this.shape = attrs.batch_input_shape && attrs.batch_input_shape.length ? attrs.batch_input_shape.slice(1) : shape
+
+    this.description = `shape: ${JSON.stringify(this.shape)}`
   }
 
   /**

src/layers/advanced_activations/ELU.js

@@ -20,6 +20,8 @@ export default class ELU extends Layer {
 
     const { alpha = 1.0 } = attrs
 
+    this.description = `alpha: ${alpha}`
+
     this.alpha = alpha
 
     // GPU setup

src/layers/advanced_activations/LeakyReLU.js

@@ -20,6 +20,8 @@ export default class LeakyReLU extends Layer {
 
     const { alpha = 0.3 } = attrs
 
+    this.description = `alpha: ${alpha}`
+
     this.alpha = alpha
 
     // GPU setup

src/layers/advanced_activations/ThresholdedReLU.js

@@ -20,6 +20,8 @@ export default class ThresholdedReLU extends Layer {
 
     const { theta = 1 } = attrs
 
+    this.description = `theta: ${theta}`
+
     this.theta = theta
 
     // GPU setup

src/layers/convolutional/Conv1D.js

@@ -31,6 +31,11 @@ export default class Conv1D extends Layer {
       use_bias = true
     } = attrs
 
+    this.description = `${filters} filters of size ${kernel_size}, striding ${strides}`
+    this.description += padding === 'valid' ? `, no border padding` : ', pad to same borders'
+    this.description += dilation_rate > 1 ? `, dilation rate ${dilation_rate}` : ''
+    this.description += activation !== 'linear' ? `, ${activation} activation` : ''
+
     if (padding !== 'valid' && padding !== 'same') {
       this.throwError('Invalid padding.')
     }

src/layers/convolutional/Conv2D.js

@@ -83,6 +83,12 @@ export default class Conv2D extends Layer {
     // Layer weights specification
     this.params = this.useBias ? ['kernel', 'bias'] : ['kernel']
 
+    this.description = `${this.kernelShape[0]} ${this.kernelShape.slice(1).join('x')} filters`
+    this.description += this.strides.some(s => s > 1) ? `, ${this.strides.join('x')} striding` : ''
+    this.description += this.padding === 'valid' ? `, no border padding` : ', pad to same borders'
+    this.description += this.dilationRate.some(r => r > 1) ? `, ${this.dilationRate.join('x')} dilation` : ''
+    this.description += this.activation !== 'linear' ? `, ${this.activation} activation` : ''
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/Conv2DTranspose.js

@@ -91,6 +91,11 @@ export default class Conv2DTranspose extends Layer {
     // Layer weights specification
     this.params = this.useBias ? ['kernel', 'bias'] : ['kernel']
 
+    this.description = `${this.kernelShape[0]} ${this.kernelShape.slice(1).join('x')} filters`
+    this.description += this.strides.some(s => s > 1) ? `, ${this.strides.join('x')} striding` : ''
+    this.description += this.padding === 'valid' ? `, no border padding` : ', pad to same borders'
+    this.description += this.activation !== 'linear' ? `, ${this.activation} activation` : ''
+
     // GPU setup
     if (this.gpu) {
       this.matMulProgram = webgl2.compileProgram(matMulProgramSource)

src/layers/convolutional/Conv3D.js

@@ -83,6 +83,12 @@ export default class Conv3D extends Layer {
     // Layer weights specification
     this.params = this.useBias ? ['kernel', 'bias'] : ['kernel']
 
+    this.description = `${this.kernelShape[0]} ${this.kernelShape.slice(1).join('x')} filters`
+    this.description += this.strides.some(s => s > 1) ? `, ${this.strides.join('x')} striding` : ''
+    this.description += this.padding === 'valid' ? `, no border padding` : ', pad to same borders'
+    this.description += this.dilationRate.some(r => r > 1) ? `, ${this.dilationRate.join('x')} dilation` : ''
+    this.description += this.activation !== 'linear' ? `, ${this.activation} activation` : ''
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/Cropping1D.js

@@ -26,6 +26,8 @@ export default class Cropping1D extends Layer {
       this.cropping = [cropping, cropping]
     }
 
+    this.description = `${JSON.stringify(this.cropping)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/Cropping2D.js

@@ -36,6 +36,8 @@ export default class Cropping2D extends Layer {
 
     this.dataFormat = data_format
 
+    this.description = `${JSON.stringify(this.cropping)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/Cropping3D.js

@@ -36,6 +36,8 @@ export default class Cropping3D extends Layer {
 
     this.dataFormat = data_format
 
+    this.description = `${JSON.stringify(this.cropping)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/SeparableConv2D.js

@@ -200,6 +200,11 @@ export default class SeparableConv2D extends Layer {
       use_bias = true
     } = attrs
 
+    this.description = `${filters} ${kernel_size.join('x')} filters, ${strides.join('x')} striding`
+    this.description += padding === 'valid' ? `, no border padding` : ', pad to same borders'
+    this.description += depth_multiplier > 1 ? `, depth multiplier: ${depth_multiplier}` : ''
+    this.description += activation !== 'linear' ? `, ${activation} activation` : ''
+
     if (Array.isArray(kernel_size)) {
       this.kernelShape = [filters, ...kernel_size]
     } else {

src/layers/convolutional/UpSampling1D.js

@@ -21,6 +21,8 @@ export default class UpSampling1D extends Layer {
     const { size = 2 } = attrs
     this.size = size
 
+    this.description = `size ${size}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/UpSampling2D.js

@@ -29,6 +29,8 @@ export default class UpSampling2D extends Layer {
 
     this.dataFormat = data_format
 
+    this.description = `size ${this.size.join('x')}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/UpSampling3D.js

@@ -29,6 +29,8 @@ export default class UpSampling3D extends Layer {
 
     this.dataFormat = data_format
 
+    this.description = `size ${this.size.join('x')}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/ZeroPadding1D.js

@@ -26,6 +26,8 @@ export default class ZeroPadding1D extends Layer {
       this.padding = [padding, padding]
     }
 
+    this.description = `${JSON.stringify(this.padding)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/ZeroPadding2D.js

@@ -36,6 +36,8 @@ export default class ZeroPadding2D extends Layer {
 
     this.dataFormat = data_format
 
+    this.description = `${JSON.stringify(this.padding)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/convolutional/ZeroPadding3D.js

@@ -36,6 +36,8 @@ export default class ZeroPadding3D extends Layer {
 
     this.dataFormat = data_format
 
+    this.description = `${JSON.stringify(this.padding)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/core/Activation.js

@@ -23,6 +23,8 @@ export default class Activation extends Layer {
     this.activation = activation
     this.activationFunc = activations[activation]
 
+    this.description = `${this.activation}`
+
     // GPU setup
     if (this.gpu) {
       this.program = webgl2.compileProgram(activationProgramSources[this.activation])

src/layers/core/Dense.js

@@ -23,6 +23,8 @@ export default class Dense extends Layer {
 
     const { units = 1, activation = 'linear', input_dim = null, use_bias = true } = attrs
 
+    this.description = `${activation} activation, output dimensions: ${units}`
+
     this.activation = activation
     this.activationFunc = activations[this.activation]
     this.units = units

src/layers/core/Dropout.js

@@ -18,6 +18,8 @@ export default class Dropout extends Layer {
 
     const { rate = 0.5 } = attrs
 
+    this.description = `${rate}`
+
     this.rate = Math.min(Math.max(0, rate), 1)
   }
 

src/layers/core/Permute.js

@@ -22,6 +22,9 @@ export default class Permute extends Layer {
     this.layerClass = 'Permute'
 
     const { dims = [] } = attrs
+
+    this.description = `${JSON.stringify(dims)}`
+
     this.dims = dims.map(dim => dim - 1)
 
     // GPU setup

src/layers/core/RepeatVector.js

@@ -24,6 +24,8 @@ export default class RepeatVector extends Layer {
     const { n = 1 } = attrs
     this.n = n
 
+    this.description = `n = ${n}`
+
     // GPU setup
     if (this.gpu) {
       this.program = webgl2.compileProgram(programSource)

src/layers/core/Reshape.js

@@ -23,6 +23,8 @@ export default class Reshape extends Layer {
     const { target_shape = [] } = attrs
     this.targetShape = target_shape
 
+    this.description = `target shape: ${JSON.stringify(this.targetShape)}`
+
     // GPU setup
     if (this.gpu) {
       this.mapInputProgram = webgl2.compileProgram(mapInputProgramSource)

src/layers/core/SpatialDropout1D.js

@@ -18,6 +18,8 @@ export default class SpatialDropout1D extends Layer {
 
     const { p = 0.5 } = attrs
 
+    this.description = `${p}`
+
     this.p = Math.min(Math.max(0, p), 1)
   }
 

src/layers/core/SpatialDropout2D.js

@@ -19,6 +19,8 @@ export default class SpatialDropout2D extends Layer {
 
     const { rate = 0.5, data_format = 'channels_last' } = attrs
 
+    this.description = `${rate}`
+
     this.rate = Math.min(Math.max(0, rate), 1)
     this.dataFormat = data_format
   }

src/layers/core/SpatialDropout3D.js

@@ -19,6 +19,8 @@ export default class SpatialDropout3D extends Layer {
 
     const { rate = 0.5, data_format = 'channels_last' } = attrs
 
+    this.description = `${rate}`
+
     this.rate = Math.min(Math.max(0, rate), 1)
     this.dataFormat = data_format
   }

src/layers/embeddings/Embedding.js

@@ -19,6 +19,8 @@ export default class Embedding extends Layer {
 
     const { input_dim = 1, output_dim = 1, input_length = 0, mask_zero = false } = attrs
 
+    this.description = `output dimensions: ${output_dim}`
+
     this.inputDim = input_dim
     this.outputDim = output_dim
     this.inputLength = input_length

src/layers/pooling/_Pooling1D.js

@@ -28,6 +28,10 @@ export default class _Pooling1D extends Layer {
     // can be `max` or `average`
     this.poolingFunc = 'max'
 
+    this.description = `pool size ${this.poolSize}`
+    this.description += this.strides > 1 ? `, striding ${this.strides} striding` : ''
+    this.description += this.padding === 'valid' ? `, no border padding` : ', pad to same borders'
+
     // GPU setup
     if (this.gpu) {
       this.poolingProgram = webgl2.compileProgram(poolingProgramSource)

src/layers/pooling/_Pooling2D.js

@@ -42,6 +42,10 @@ export default class _Pooling2D extends Layer {
     // can be `max` or `average`
     this.poolingFunc = 'max'
 
+    this.description = `pool size ${this.poolSize.join('x')}`
+    this.description += this.strides.some(s => s > 1) ? `, ${this.strides.join('x')} striding` : ''
+    this.description += this.padding === 'valid' ? `, no border padding` : ', pad to same borders'
+
     // GPU setup
     if (this.gpu) {
       this.poolingProgram = webgl2.compileProgram(poolingProgramSource)

src/layers/pooling/_Pooling3D.js

@@ -42,6 +42,10 @@ export default class _Pooling3D extends Layer {
     // can be `max` or `average`
     this.poolingFunc = 'max'
 
+    this.description = `pool size ${this.poolSize.join('x')}`
+    this.description += this.strides.some(s => s > 1) ? `, ${this.strides.join('x')} striding` : ''
+    this.description += this.padding === 'valid' ? `, no border padding` : ', pad to same borders'
+
     // GPU setup
     if (this.gpu) {
       this.poolingProgram = webgl2.compileProgram(poolingProgramSource)

src/layers/recurrent/GRU.js

@@ -61,6 +61,14 @@ export default class GRU extends Layer {
     // Layer weights specification
     this.params = this.use_bias ? ['kernel', 'recurrent_kernel', 'bias'] : ['kernel', 'recurrent_kernel']
 
+    this.description = `output dimensions: ${this.units}`
+    this.description += this.activation !== 'linear' ? `, ${this.activation} activation` : ''
+    this.description +=
+      this.recurrentActivation !== 'linear' ? `, ${this.recurrentActivation} recurrent activation` : ''
+    this.description += this.returnSequences ? `, return sequences` : ''
+    this.description += this.goBackwards ? `, backward direction` : ''
+    this.description += this.stateful ? `, stateful` : ''
+
     // GPU setup
     if (this.gpu) {
       this.copyTextureProgram = webgl2.compileProgram(copyTextureProgramSource)

src/layers/recurrent/LSTM.js

@@ -61,6 +61,14 @@ export default class LSTM extends Layer {
     // Layer weights specification
     this.params = this.use_bias ? ['kernel', 'recurrent_kernel', 'bias'] : ['kernel', 'recurrent_kernel']
 
+    this.description = `output dimensions: ${this.units}`
+    this.description += this.activation !== 'linear' ? `, ${this.activation} activation` : ''
+    this.description +=
+      this.recurrentActivation !== 'linear' ? `, ${this.recurrentActivation} recurrent activation` : ''
+    this.description += this.returnSequences ? `, return sequences` : ''
+    this.description += this.goBackwards ? `, backward direction` : ''
+    this.description += this.stateful ? `, stateful` : ''
+
     // GPU setup
     if (this.gpu) {
       this.copyTextureProgram = webgl2.compileProgram(copyTextureProgramSource)

src/layers/recurrent/SimpleRNN.js

@@ -55,6 +55,12 @@ export default class SimpleRNN extends Layer {
     // Layer weights specification
     this.params = this.use_bias ? ['kernel', 'recurrent_kernel', 'bias'] : ['kernel', 'recurrent_kernel']
 
+    this.description = `output dimensions: ${this.units}`
+    this.description += this.activation !== 'linear' ? `, ${this.activation} activation` : ''
+    this.description += this.returnSequences ? `, return sequences` : ''
+    this.description += this.goBackwards ? `, backward direction` : ''
+    this.description += this.stateful ? `, stateful` : ''
+
     // GPU setup
     if (this.gpu) {
       this.copyTextureProgram = webgl2.compileProgram(copyTextureProgramSource)