Dtype object is not a valid JAX array type. Only arrays of numeric types are supported by JAX. about lightweight_mmm HOT 5 OPEN

have you transformed your dataframe or dataset to array, e.g. with np.array(media_data_train)?

from lightweight_mmm.

I did like this, on the tutorial. it is an array but still doesn't let me use the Scalers:

from lightweight_mmm.

it is an array, and is float32, it should be in good shape already. WHich dataset threw an error? target? media? or others?

from lightweight_mmm.

As you can see below the target_train caused errors. The Scalers & fit_transofrm of media_data_train and costs worked.

And this is how target_train looks like:

from lightweight_mmm.

Hmmmm.... I know you have checked it already, but would there be any chance that your target_train has nan, null, zero, or different data dtype?

Do you mind to provide your notebook, and data for further investigation? I am not sure if there is any pm function in github.

from lightweight_mmm.