Skip to content

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

@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
you-n-g pushed a commit to you-n-g/qlib that referenced this pull request Aug 16, 2025
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