Custom Activation function

[zszazi]

Hey
I am having this custom activation function which is not listed in the replay.ACTIVATIONS nor in the keras package
Although I am able to Train the Keras model with custom activation function, How do I generate the plot for custom activation function in deepreplay?

on running the below cell

replay = Replay(replay_filename='spambase_dataset.h5', group_name=group_name)

fs = replay.build_feature_space(ax_fs, layer_name='hidden', scale_fixed=False)
ph = replay.build_probability_histogram(ax_ph_neg, ax_ph_pos)
lh = replay.build_loss_histogram(ax_lh)
lm = replay.build_loss_and_metric(ax_lm, 'acc')

I get the following error

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-31-f06018e9e1b6> in <module>()
      1 
----> 2 replay = Replay(replay_filename='spambase_dataset.h5', group_name=group_name)
      3 
      4 fs = replay.build_feature_space(ax_fs, layer_name='hidden', scale_fixed=False)
      5 ph = replay.build_probability_histogram(ax_ph_neg, ax_ph_pos)

15 frames
/usr/local/lib/python3.6/dist-packages/keras/utils/generic_utils.py in deserialize_keras_object(identifier, module_objects, custom_objects, printable_module_name)
    165             if fn is None:
    166                 raise ValueError('Unknown ' + printable_module_name +
--> 167                                  ':' + function_name)
    168         return fn
    169     else:

ValueError: Unknown activation function:swish