How to get label when the dataset is TSDatasetH?

[LUS8806]

Hi,
I'm trying to get the label data from the dataset when the dataset is TSDatasetH.

When I do using these codes:

label_df = dataset.prepare("test", col_set="label")

It returns a qlib.data.dataset.TSDataSampler.

I need a dataframe. How can I get label dataframe when the dataset is TSDatasetH?

Thx,

Song

[you-n-g]

https://github.com/microsoft/qlib/blob/main/tests/test_dataset.py#L46
Here is an example of using the data sampler returned by TSDatasetH.
You can get a time series of data based on the label col_set.

Here is another example of getting label using TSDatasetH

[github-actions]

This issue is stale because it has been open for three months with no activity. Remove the stale label or comment on the issue otherwise this will be closed in 5 days

[zoaamour]

label_df = dataset.handler.fetch(dataset.segments["test"], col_set="label")

@LUS8806