Output.ftl 478 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11

<#if element.softmaxOutput>
        ${element.name} = mx.symbol.SoftmaxOutput(data=${element.inputs[0]},
            name="${element.name}")
<#elseif element.logisticRegressionOutput>
        ${element.name} = mx.symbol.LogisticRegressionOutput(data=${element.inputs[0]},
            name="${element.name}")
<#elseif element.linearRegressionOutput>
        ${element.name} = mx.symbol.LinearRegressionOutput(data=${element.inputs[0]},
            name="${element.name}")
</#if>