emadeldeen24/AttnSleep

question about the Class-Aware Loss Function and Optimization

MatrixWP opened this issue · 1 comments

MatrixWP commented

In your paper, the weight of classes is , but in your code, it seems like the formula is(N is the number of classes), I wonder if the code below is right?

AttnSleep/utils/util.py

Line 73 in eb2beeb

class_weight[key] = round(class_weight[key] * factor, 2)

emadeldeen24 commented

I updated it. Thank you for your comment.