MLFlow Output Signature
Options
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
12-14-2023 07:34 AM
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]") ])