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

refactor TRA #531

Merged
merged 8 commits into from
Jul 30, 2021
Merged

refactor TRA #531

merged 8 commits into from
Jul 30, 2021

Conversation

evanzd
Copy link
Contributor

@evanzd evanzd commented Jul 21, 2021

Refactor TRA to support using Qlib dataset.

Motivation and Context

The current implementation cannot support Qlib dataset like Alpha158 and Alpha360. In this PR, we have improved MTSDatasetH that is compatible to these datasets. Besides, we also support transport samples as daily groups and support oracle transport that discovers the optimal assignment for model diversity.

How Has This Been Tested?

Has been tested by running the accompanied config files: workflow_config_tra_Alpha360.yaml, workflow_config_tra_Alpha158.yaml and workflow_config_tra_Alpha158_full.yaml .

Screenshots of Test Results (if appropriate):

Running workflow_config_tra_Alpha360.yaml with seed = 6000:
image

Types of changes

  • Fix bugs
  • Add new feature
  • Update documentation

qlib/contrib/data/dataset.py Outdated Show resolved Hide resolved
@evanzd evanzd marked this pull request as ready for review July 28, 2021 06:37
@evanzd
Copy link
Contributor Author

evanzd commented Jul 28, 2021

@you-n-g Please help review this PR at your convenience. Thanks :)

@evanzd evanzd changed the title refactor TRA [work in progress] refactor TRA Jul 28, 2021
@liufeirong
Copy link

In the predict function,it called the test_epoch method which used the label col.In the truth, when predict, the label is None

@you-n-g you-n-g merged commit da1f4db into microsoft:main Jul 30, 2021
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.

3 participants