Enhance predict_fn in init_model to support multiple output columns

validmind-library
2.8.29
documentation
Published

July 23, 2025

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.