コード例 #1
0
ファイル: MacroUtils.cs プロジェクト: zafarsk/machinelearning
        internal static void ConvertIPredictorModelsToArray(IHostEnvironment env, RunContext context, List <EntryPointNode> subGraphNodes,
                                                            Var <PredictorModel>[] predModelVars, string outputVarName)
        {
            var predictorArrayConverterArgs = new ArrayIPredictorModelInput();
            var inputBindingMap             = new Dictionary <string, List <ParameterBinding> >();
            var inputMap = new Dictionary <ParameterBinding, VariableBinding>();

            var argName = nameof(predictorArrayConverterArgs.Model);

            inputBindingMap.Add(argName, new List <ParameterBinding>());
            for (int i = 0; i < predModelVars.Length; i++)
            {
                var paramBinding = new ArrayIndexParameterBinding(argName, i);
                inputBindingMap[argName].Add(paramBinding);
                inputMap[paramBinding] = new SimpleVariableBinding(predModelVars[i].VarName);
            }
            var outputMap = new Dictionary <string, string>();
            var output    = new ArrayVar <PredictorModel>();

            outputMap.Add(nameof(MacroUtils.ArrayIPredictorModelOutput.OutputModel), outputVarName);
            var arrayConvertNode = EntryPointNode.Create(env, "Data.PredictorModelArrayConverter", predictorArrayConverterArgs,
                                                         context, inputBindingMap, inputMap, outputMap);

            subGraphNodes.Add(arrayConvertNode);
        }
コード例 #2
0
ファイル: MacroUtils.cs プロジェクト: zafarsk/machinelearning
        public static ArrayIPredictorModelOutput MakeArray(IHostEnvironment env, ArrayIPredictorModelInput input)
        {
            var result = new ArrayIPredictorModelOutput
            {
                OutputModel = input.Model
            };

            return(result);
        }