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

Forecasting: generate time series as cohorts #2355

Merged
merged 15 commits into from
Oct 3, 2023

Conversation

romanlutz
Copy link
Contributor

Description

With raiwidgets we can inject time series as prebuilt cohorts, but in Studio we don't have that option, so this PR adds the functionality to generate time series as cohorts if they weren't provided. This only happens if time series ID features were provided, otherwise we assume that there is only a single time series in the dataset.

While working on this I also uncovered a bug in the function IsTimeOrTimeSeriesIdColumn. Checking for datetime features doesn't quite make sense since we populate the index of the dataset with the time column that we get from Python. I removed that check since it actually broke the single time series case.

Overall, I also added two new test scenarios that capture the time series generation cases.

There are no UI changes.

Checklist

  • I have added screenshots above for all UI changes.
  • I have added e2e tests for all UI changes.
  • Documentation was updated if it was needed.

@codecov-commenter
Copy link

codecov-commenter commented Sep 22, 2023

Codecov Report

Merging #2355 (ee15929) into main (5833a88) will not change coverage.
The diff coverage is n/a.

❗ Current head ee15929 differs from pull request most recent head 67978af. Consider uploading reports for the commit 67978af to get more accurate results

@@           Coverage Diff           @@
##             main    #2355   +/-   ##
=======================================
  Coverage   92.40%   92.40%           
=======================================
  Files         108      108           
  Lines        5415     5415           
=======================================
  Hits         5004     5004           
  Misses        411      411           
Flag Coverage Δ
unittests 92.40% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

2 similar comments
1 similar comment
@romanlutz romanlutz merged commit 426c842 into main Oct 3, 2023
@romanlutz romanlutz deleted the romanlutz/forecasting_cohort_gen branch October 3, 2023 16:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants