Trying to use fairlean mitigation with a pyTorch regression model. Create a dataset that outputs X and Y DataLoader the X and Y Record A = X['senstiveFeatures'] Fix test dataset.