class_to_idx parameter should not be a required parameter for regression tasks
baselmousi opened this issue · comments
Basel Mousi commented
Hello,
I was trying to get the top neurons associated with a linear regression probe on a set of synthetic data. I obtained that the class_to_idx is a required positional argument; Shouldn't it be optional for regression tasks?
Thanks in advance