coremltools
coremltools copied to clipboard
Update API: utils.evaluate_regressor and utils.evaluate_classifier
Don't assume source predictions are always stored (in the data frame) in a column called prediction. Instead, assume that column name matches the key in the dictionary returned by predict, similar to how we assume input keys to predict match other columns. This parameter value defaults to target.
This is an API breaking change.
Fixes #58
CI run: https://gitlab.com/coremltools1/coremltools/-/pipelines/608715939
Updated CI: https://gitlab.com/coremltools1/coremltools/-/pipelines/611972518