Enhance predict_fn
in init_model
to support multiple output columns
validmind-library
2.8.29
documentation
This update enhances the predict_fn
callable function parameter in init_model
to support multiple output columns. You can now return either a single value or a dictionary from the function. If a dictionary is returned, the key named prediction
will be used as the prediction column. Additional keys in the dictionary can be included as extra columns in the dataset object.