ch-sims-v2 icon indicating copy to clipboard operation
ch-sims-v2 copied to clipboard

can't find dataset['train_mix']

Open wx3771 opened this issue 3 years ago • 3 comments

image Could you please provide this ? Thank you!

wx3771 avatar Sep 20 '22 01:09 wx3771

https://github.com/thuiar/ch-sims-v2/blob/6eb09e5e9d17355cec436c59f4d3fdb41f3b5891/data/load_data.py#L40

train_mix is an augmentation from the original train set. Using the following operations.

Columbine21 avatar Sep 20 '22 01:09 Columbine21

q1: when args.supvised_nums default value is 2722, these code will never execute image

q2: no mask image

wx3771 avatar Sep 20 '22 03:09 wx3771

image

Could you please provide this ? Thank you!

For the convenience of downloading the newly provided dataset, we divided the supervised data and unsupervised data into two ".pkl" files, and "train_mix” will put the supervised data and unsupervised data together. You can concatenate supervised and unsupervised data together as "train_mix" data. In addition, "mask" is set to 1 for supervised data and 0 for unsupervised data. For example, the data corresponding to MASK: [1,1,0,0] is Data: [supervised data, supervised data, unsupervised data, unsupervised data].

lyh512796310 avatar Sep 20 '22 03:09 lyh512796310