Skip to content

Commit cd67f44

Browse files
authored
Keras adapter with uni-dimensional targets (#266)
* support uni-dimensional array * pin docutils
1 parent afa75b7 commit cd67f44

2 files changed

Lines changed: 6 additions & 1 deletion

File tree

mlprimitives/adapters/keras.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,11 @@ def _setdefault(self, kwargs, key, value):
8787

8888
def _augment_hyperparameters(self, X, mode, kwargs):
8989
shape = np.asarray(X)[0].shape
90-
length = shape[0]
90+
if shape:
91+
length = shape[0]
92+
else:
93+
length = 1 # supporting shape (l, )
94+
9195
self._setdefault(kwargs, '{}_shape'.format(mode), shape)
9296
self._setdefault(kwargs, '{}_dim'.format(mode), length)
9397
self._setdefault(kwargs, '{}_length'.format(mode), length)

setup.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@
5353
'm2r>=0.2.0,<0.3',
5454
'Sphinx>=1.7.1,<3',
5555
'sphinx_rtd_theme>=0.2.4,<0.5',
56+
'docutils>=0.14,<0.18',
5657
'ipython>=6.5.0',
5758

5859
# style check

0 commit comments

Comments
 (0)