コード例 #1
0
ファイル: Linear.cs プロジェクト: ygambhir/OpenMined
        public override JToken GetConfig()
        {
            var config = new JObject
            {
                { "name", "linear" },
                { "trainable", true },
                { "dtype", "float32" },
                { "output", _output },
                { "input", _input },
                { "bias", _bias?.GetConfig() },
                { "weights", _weights?.GetConfig() },
                { "activation", "linear" },
                { "use_bias", true },
                {
                    "kernel_initializer", new JObject
                    {
                        { "class_name", "VarianceScaling" },
                        {
                            "config", new JObject
                            {
                                { "scale", 1.0 },
                                { "mode", "fan_avg" },
                                { "distribution", "uniform" },
                                { "seed", null }
                            }
                        }
                    }
                },
                {
                    "bias_initializer", new JObject
                    {
                        { "class_name", "Zeros" },
                        { "config", new JObject() }
                    }
                },
                { "kernel_regularizer", null },
                { "bias_regularizer", null },
                { "activity_regularizer", null },
                { "kernel_constraint", null },
                { "bias_constraint", null }
            };

            return(config);
        }