[
https://issues.apache.org/jira/browse/SPARK-39058?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17529462#comment-17529462
]
Weichen Xu commented on SPARK-39058:
------------------------------------
CC [~ruifengz] Are you interested in contributing this feature ? :)
> Add `getInputSignature` and `getOutputSignature` APIs for spark ML
> models/transformers
> --------------------------------------------------------------------------------------
>
> Key: SPARK-39058
> URL: https://issues.apache.org/jira/browse/SPARK-39058
> Project: Spark
> Issue Type: New Feature
> Components: ML
> Affects Versions: 3.4.0
> Reporter: Weichen Xu
> Priority: Major
>
> Add `getInputSignature` and `getOutputSignature` APIs for spark ML
> models/transformers:
> The `getInputSignature` API return a list of column name and type which
> represent the ML model / transformer's input columns.
> The `getOutputSignature` API return a list of column column name and type
> which represent the ML model / transformer's output columns.
> These 2 APIs are useful to in third-party library such as mlflow, which
> requires the information of input / output signature of the Model.
--
This message was sent by Atlassian Jira
(v8.20.7#820007)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
