MLFlow Output Signature

pablobd — Thu, 14 Dec 2023 15:34:51 GMT

I want to define the output of a MLFlow model to include a dictionary of string to float. I did this, but it doesn't work. Do you know if it's possible at all?

from mlflow.types.schema import Schema, ColSpec

output_schema = Schema([ ColSpec("dictionary_output", "map[string,float]") ])

Re: MLFlow Output Signature

pablobd — Mon, 18 Dec 2023 18:16:50 GMT

Solved with the json module that converts the dict into string, thanks! 🙂

import json

json.dumps(...)

topic Re: MLFlow Output Signature in Machine Learning

MLFlow Output Signature

Re: MLFlow Output Signature