I want to create a custom loss function for a regression model which takes an additional feature (which is not an input) as an argument. The dataset is of the form X, Y and