Skip to content
This repository was archived by the owner on Jun 23, 2025. It is now read-only.

Commit ad247c5

Browse files
SriRangaTarungabrieldemarmiesse
authored andcommitted
Fix Capsule docs example and remove unused import (#434)
* Fix docs error * Remove unused import
1 parent ab9d90b commit ad247c5

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

keras_contrib/layers/capsule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ class Capsule(Layer):
3939
is generally used with the 'squash' activation function
4040
(recommended). To use the squash activation function, do :
4141
42-
from keras_contrib import squash
42+
from keras_contrib.activations import squash
4343
4444
capsule = Capsule(num_capsule=10,
4545
dim_capsule=10,

tests/keras_contrib/layers/test_capsule.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from keras_contrib.utils.test_utils import layer_test
66
from keras import backend as K
7-
from keras_contrib import backend as KC
87
from keras_contrib.layers import capsule
98
from keras.models import Sequential
109

0 commit comments

Comments
 (0)