alihan/ml_things
1
0
Fork 0
mirror of https://github.com/gmihaila/ml_things.git synced 2021-10-04 01:29:04 +03:00
Code Issues Projects Releases Wiki Activity

Added support input for numpy array not just list

Browse Source
This commit is contained in:
George Mihaila
2020-09-06 09:56:34 -05:00
committed by GitHub
parent 932330af6a
commit 4dd59b429d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
padd_array/README.md
View File

@@ -49,7 +49,8 @@ def pad_numpy(variable_length_array, fixed_length=None, axis=1):
fixed_length = max([len(row) for row in variable_length_array]) if fixed_length is None else fixed_length
# array of arrays
if isinstance(variable_length_array[0], list):
if isinstance(variable_length_array[0], list) or isinstance(
variable_length_array[0], np.ndarray):
if axis == 1:
# perform pading on rows
@@ -72,7 +73,8 @@ def pad_numpy(variable_length_array, fixed_length=None, axis=1):
return numpy_array
# array of values
elif isinstance(variable_length_array, list):
elif isinstance(variable_length_array, list) or isinstance(
variable_length_array, np.ndarray)::
if axis == 1:
# perform pading on rows