I\'m trying to replicate the AIX360 ProfWeight Explainer training, where the authors on the GitHub example use the following code:
from aix360.datasets.cifar_