wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-06-27 16:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Import on top of file

Browse Source
This commit is contained in:
Fariz Rahman
2017-01-26 11:37:40 +05:30
committed by GitHub
parent 0163830c92
commit 1be552eb6d
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/layers/test_advanced_activations.py
+3 -5
View File
@@ -1,19 +1,17 @@
import pytest
from keras.utils.test_utils import layer_test, keras_test
from keras_contrib import layers
from keras_contrib.layers import advanced_activations
@keras_test
def test_pelu():
from keras_contrib.layers.advanced_activations import PELU
layer_test(PELU, kwargs={},
layer_test(advanced_activations.PELU, kwargs={},
input_shape=(2, 3, 4))
@keras_test
def test_pelu_share():
from keras_contrib.layers.advanced_activations import PELU
layer_test(PELU, kwargs={'shared_axes': 1},
layer_test(advanced_activations.PELU, kwargs={'shared_axes': 1},
input_shape=(2, 3, 4))