wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-14 12:30:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix serialization

Browse Source
This commit is contained in:
farizrahman4u
2017-03-19 09:25:58 +05:30
parent 80c34ac886
commit 2dc2b78213
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/layers/core.py
+5 -5
View File
@@ -89,12 +89,12 @@ class CosineDense(Layer):
the output would have shape `(nb_samples, units)`.
"""
def __init__(self, units, init='glorot_uniform',
def __init__(self, units, kernel_initializer='glorot_uniform',
activation=None, weights=None,
kernel_regularizer=None, bias_regularizer=None, activity_regularizer=None,
kernel_constraint=None, bias_constraint=None,
use_bias=True, input_dim=None, **kwargs):
self.init = initializers.get(init)
self.kernel_initializer = initializers.get(kernel_initializer)
self.activation = activations.get(activation)
self.units = units
self.input_dim = input_dim
@@ -122,7 +122,7 @@ class CosineDense(Layer):
ndim=2)]
self.kernel = self.add_weight((input_dim, self.units),
initializer=self.init,
initializer=self.kernel_initializer,
name='{}_W'.format(self.name),
regularizer=self.kernel_regularizer,
constraint=self.kernel_constraint)
@@ -165,8 +165,8 @@ class CosineDense(Layer):
def get_config(self):
config = {'units': self.units,
'init': self.init.__name__,
'activation': self.activation.__name__,
'kernel_initializer': initializers.serialize(self.kernel_initializer),
'activation': activations.serialize(self.activation),
'kernel_regularizer': regularizers.serialize(self.kernel_regularizer),
'bias_regularizer': regularizers.serialize(self.bias_regularizer),
'activity_regularizer': regularizers.serialize(self.activity_regularizer),