diff --git a/src/layers/advanced_activations/ELU.js b/src/layers/advanced_activations/ELU.js index 89982df..0c2acaa 100644 --- a/src/layers/advanced_activations/ELU.js +++ b/src/layers/advanced_activations/ELU.js @@ -9,8 +9,8 @@ export default class ELU extends Layer { * Creates a ELU activation layer * @param {number} alpha - scale for the negative factor */ - constructor (alpha = 1.0) { - super({}) + constructor (alpha = 1.0, attrs = {}) { + super(attrs) this.alpha = alpha } diff --git a/src/layers/advanced_activations/LeakyReLU.js b/src/layers/advanced_activations/LeakyReLU.js index a18ce88..6cd6f8a 100644 --- a/src/layers/advanced_activations/LeakyReLU.js +++ b/src/layers/advanced_activations/LeakyReLU.js @@ -9,8 +9,8 @@ export default class LeakyReLU extends Layer { * Creates a LeakyReLU activation layer * @param {number} alpha - negative slope coefficient */ - constructor (alpha = 0.3) { - super({}) + constructor (alpha = 0.3, attrs = {}) { + super(attrs) this.alpha = alpha } diff --git a/src/layers/advanced_activations/PReLU.js b/src/layers/advanced_activations/PReLU.js index 43e9b83..b9a3f25 100644 --- a/src/layers/advanced_activations/PReLU.js +++ b/src/layers/advanced_activations/PReLU.js @@ -14,8 +14,8 @@ export default class PReLU extends Layer { /** * Creates a PReLU activation layer */ - constructor () { - super({}) + constructor (attrs = {}) { + super(attrs) // Layer weights specification this.params = ['alphas'] diff --git a/src/layers/advanced_activations/ParametricSoftplus.js b/src/layers/advanced_activations/ParametricSoftplus.js index 27fced0..ec8fc15 100644 --- a/src/layers/advanced_activations/ParametricSoftplus.js +++ b/src/layers/advanced_activations/ParametricSoftplus.js @@ -9,8 +9,8 @@ export default class ParametricSoftplus extends Layer { /** * Creates a ParametricSoftplus activation layer */ - constructor () { - super({}) + constructor (attrs = {}) { + super(attrs) // Layer weights specification this.params = ['alphas', 'betas'] diff --git a/src/layers/advanced_activations/SReLU.js b/src/layers/advanced_activations/SReLU.js index 69f5ec9..c6bb578 100644 --- a/src/layers/advanced_activations/SReLU.js +++ b/src/layers/advanced_activations/SReLU.js @@ -9,8 +9,8 @@ export default class SReLU extends Layer { /** * Creates a SReLU activation layer */ - constructor () { - super({}) + constructor (attrs = {}) { + super(attrs) // Layer weights specification this.params = ['t_left', 'a_left', 't_right', 'a_right'] diff --git a/src/layers/advanced_activations/ThresholdedReLU.js b/src/layers/advanced_activations/ThresholdedReLU.js index 8da41b3..6f8de47 100644 --- a/src/layers/advanced_activations/ThresholdedReLU.js +++ b/src/layers/advanced_activations/ThresholdedReLU.js @@ -9,8 +9,8 @@ export default class ThresholdedReLU extends Layer { * Creates a ThresholdedReLU activation layer * @param {number} theta - float >= 0. Threshold location of activation. */ - constructor (theta = 1.0) { - super({}) + constructor (theta = 1.0, attrs = {}) { + super(attrs) this.theta = theta } diff --git a/src/layers/core/Activation.js b/src/layers/core/Activation.js index 2e850b3..e6e4f9f 100644 --- a/src/layers/core/Activation.js +++ b/src/layers/core/Activation.js @@ -10,7 +10,7 @@ export default class Activation extends Layer { * @param {string} activation - name of activation function */ constructor (activation, attrs = {}) { - super({}) + super(attrs) this.activation = activations[activation] } diff --git a/src/layers/core/Dropout.js b/src/layers/core/Dropout.js index 7fd5c1d..a835505 100644 --- a/src/layers/core/Dropout.js +++ b/src/layers/core/Dropout.js @@ -9,8 +9,8 @@ export default class Dropout extends Layer { * Creates an Dropout layer * @param {number} p - fraction of the input units to drop (between 0 and 1) */ - constructor (p) { - super({}) + constructor (p, attrs = {}) { + super(attrs) this.p = Math.min(Math.max(0, p), 1) } diff --git a/src/layers/core/Flatten.js b/src/layers/core/Flatten.js index 9383b27..993736a 100644 --- a/src/layers/core/Flatten.js +++ b/src/layers/core/Flatten.js @@ -12,8 +12,8 @@ export default class Flatten extends Layer { /** * Creates a Flatten layer */ - constructor () { - super({}) + constructor (attrs = {}) { + super(attrs) } /** diff --git a/src/layers/core/Permute.js b/src/layers/core/Permute.js index 7b98aa9..b182dfe 100644 --- a/src/layers/core/Permute.js +++ b/src/layers/core/Permute.js @@ -10,8 +10,8 @@ export default class Permute extends Layer { * Creates a Permute layer * @param {number[]} dims */ - constructor (dims) { - super({}) + constructor (dims, attrs = {}) { + super(attrs) this.dims = dims.map(dim => dim - 1) } diff --git a/src/layers/core/RepeatVector.js b/src/layers/core/RepeatVector.js index 92ba064..12e52cb 100644 --- a/src/layers/core/RepeatVector.js +++ b/src/layers/core/RepeatVector.js @@ -12,8 +12,8 @@ export default class RepeatVector extends Layer { * Creates a RepeatVector layer * @param {number} n */ - constructor (n) { - super({}) + constructor (n, attrs = {}) { + super(attrs) this.n = n } diff --git a/src/layers/core/Reshape.js b/src/layers/core/Reshape.js index 0fe4b06..0ad607b 100644 --- a/src/layers/core/Reshape.js +++ b/src/layers/core/Reshape.js @@ -13,8 +13,8 @@ export default class Reshape extends Layer { * Creates a Reshape layer * @param {number[]} shape */ - constructor (shape) { - super({}) + constructor (shape, attrs = {}) { + super(attrs) this.shape = shape }