Closed sarahmish closed 2 years ago
In the current implementation of the keras adapter, we set that input of variable dimension. However, we do not support the same feature for the target size. https://github.com/MLBazaar/MLPrimitives/blob/9195968a1377de37ef87f15952a67b587200a6d2/mlprimitives/adapters/keras.py#L98
keras
We propose to apply the same augmentation for the target argument by passing y to the augment function.
target
y
Description
In the current implementation of the
keras
adapter, we set that input of variable dimension. However, we do not support the same feature for the target size. https://github.com/MLBazaar/MLPrimitives/blob/9195968a1377de37ef87f15952a67b587200a6d2/mlprimitives/adapters/keras.py#L98We propose to apply the same augmentation for the
target
argument by passingy
to the augment function.