MLBazaar/MLPrimitives

Dynamic `target_shape` in `keras` adapter

sarahmish opened this issue · 0 comments

sarahmish commented
  • MLPrimitives version: 0.2.5
  • Python version: 3.7
  • Operating System: macOS 11.5.1

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.

MLPrimitives/mlprimitives/adapters/keras.py

Line 98 in 9195968

self._augment_hyperparameters(X, kwargs)

We propose to apply the same augmentation for the target argument by passing y to the augment function.