Is the model training is wrong?

[eardic]

TMC-Shapley algorithm in the paper says that the model is trained for each new datum that is added to the batch. However, in the line below the model is trained only when the batch size is the same as the test size. Why is that so?

DataShapley/DShap.py

Line 365 in 96e8ecb

or len(set(y_batch)) == len(set(self.y_test))): ##FIXIT

Thanks.

[camhpj]

This is a check to ensure that the batch contains both positive and negative labels. Whatever model you are using will error if only 1 unique label is present.