Tian0426/CL-HAR

Report a Bug in main_supervised_baseline.py

xiqxin1 opened this issue · 2 comments

Hi Tian,

Thank you very much for sharing the code.

In lines 167-168, you add:

trgs = torch.cat((trgs, target))#

feats = torch.cat((feats, features), 0)#

which makes the length of trgs become two times as prds when calculating the confusion_matrix(in line 175). I'm not sure if it is correct.

Hi @xiqxin1,

Thank you for pointing this out. You can delete these two lines in your code. Correction has been made in the repo!

Thank you!