How to reproduce the error:
Create a new small dataset of max 5 rows with only 2 labels and train a model where it predicts always the one value out of 2.
How to fix the error:
(There are 2 parts that need to be fixed)
In Line 67 in analytics/models/pipeline need to change the position for calculation of the ROC score
Based on the documentation of the package:
Also at the splitting part to train and test set, need to ensure that each one of the classes will always be represented into the test side, otherwise, the same error will occur again.
The error is:
How to reproduce the error: Create a new small dataset of max 5 rows with only 2 labels and train a model where it predicts always the one value out of 2.
How to fix the error: (There are 2 parts that need to be fixed)
In Line 67 in analytics/models/pipeline need to change the position for calculation of the ROC score
Based on the documentation of the package:
Also at the splitting part to train and test set, need to ensure that each one of the classes will always be represented into the test side, otherwise, the same error will occur again.