Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Categorical column after sequence_index column #314

Closed
fealho opened this issue Feb 10, 2021 · 0 comments · Fixed by #357
Closed

Categorical column after sequence_index column #314

fealho opened this issue Feb 10, 2021 · 0 comments · Fixed by #357
Assignees
Labels
bug Something isn't working
Milestone

Comments

@fealho
Copy link
Member

fealho commented Feb 10, 2021

Currently, if a dataset contains a categorical column immediately after the sequence_index column the fit breaks with the error TypeError: cannot perform reduce with flexible type. Below is an example that reproduces the issue:

from sdv.demo import load_timeseries_demo
from sdv.timeseries import PAR
import pandas as pd

data = load_timeseries_demo()
data.insert(loc=2, column="Categorical", value="a","b"*(data.shape0//2))
print(data)

model = PAR(
    entity_columns = 'Symbol',
    context_columns = 'MarketCap', 'Sector', 'Industry',
    sequence_index='Date'
)

model.fit(data)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants