[ https://issues.apache.org/jira/browse/SPARK-39058?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17531100#comment-17531100 ]
zhengruifeng commented on SPARK-39058: -------------------------------------- [~weichenxu123] I can help reivew. BTW, is there some background or design on this work? are those api implemented in some counterpart libraries like scikit-learn? > 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