AI开发平台ModelArts-自定义算子代码模板和规范:自定义算子代码模板

时间：2023-11-01 16:20:20

AI开发平台ModelArts

自定义算子代码模板

新建自定义算子时，MLS Editor提供了代码模板，方便用户高效开发算子。

class MLSClassName:    # init parameters    def __init__(self,                 inputs,                 param_1,                 param_2):        self.inputs = inputs        self.param_1 = param_1        self.param_2 = param_2        self.upper_output = None        self._outputs = {}    # core code for customized algorithm    def run(self):        # get upper output of workflow        self.upper_output = self.inputs["upper_output"]        # ...core code...        # output format        self._outputs = {            "output_port_1": "output_result"        }    # user called method for getting algorithm result    def get_outputs(self):        return self._outputs# call form for algorithminputs = {    "upper_output": None  #@input {"type":"DataFrame"}}params = {    "inputs": inputs,    "param_1": "param_value_1",  #@param {"label":"param_1","type":"string","required":"false","helpTip":""}    "param_2": "param_value_2",  #@param {"label":"param_1","type":"enum","options":"one,two,three","required":"true","helpTip":""}    "param_3": "param_value_3",  #@param {"label":"param_1","type":"integer","range":"(0,none)","required":"true","helpTip":""}    "param_4": "param_value_4"  #@param {"label":"param_1","type":"number","range":"(0,1)","required":"true","helpTip":""}}mls_instance_#id# = MLSClassName(**params)mls_instance_#id#.run()#@output {"label":"dataframe","name":"mls_instance_#id#.get_outputs()['output_port_1']","type":"DataFrame"}

上一篇：AI开发平台ModelArts-查看算子:自定义算子

下一篇：AI开发平台ModelArts-派生:样例