Report a Bug in main_supervised_baseline.py

[xiqxin1]

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.

[Tian0426]

Hi @xiqxin1,

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

[xiqxin1]

Thank you!