【联邦学习FATE框架实战】(三)MNIST神经网络(Keras)

时间:2025-04-12 11:13:21
{ "dsl_version": 2, "initiator": { "role": "guest", "party_id": 10000 }, "role": { "arbiter": [10000], "host": [10000], "guest": [10000] }, "job_parameters": { "common": { "work_mode": 0, "backend": 0 } }, "component_parameters": { "common": { "dataio_0": { "with_label": true }, "homo_nn_0": { "encode_label":true, "max_iter": 20, "batch_size": -1, "early_stop": { "early_stop": "diff", "eps": 0.0001 }, "optimizer": { "learning_rate": 0.0015, "decay": 0.0, "beta_1": 0.9, "beta_2": 0.999, "epsilon": 1e-07, "amsgrad": false, "optimizer": "Adam" }, "loss": "categorical_crossentropy", "metrics": ["accuracy", "AUC"], "nn_define": {"class_name": "Sequential", "config": {"name": "sequential", "layers": [{"class_name": "InputLayer", "config": {"batch_input_shape": [null, 784], "dtype": "float32", "sparse": false, "ragged": false, "name": "dense_input"}}, {"class_name": "Dense", "config": {"name": "dense", "trainable": true, "batch_input_shape": [null, 784], "dtype": "float32", "units": 512, "activation": "relu", "use_bias": true, "kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Dense", "config": {"name": "dense_1", "trainable": true, "dtype": "float32", "units": 256, "activation": "relu", "use_bias": true, "kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}, {"class_name": "Dense", "config": {"name": "dense_2", "trainable": true, "dtype": "float32", "units": 10, "activation": "softmax", "use_bias": true, "kernel_initializer": {"class_name": "GlorotUniform", "config": {"seed": null}}, "bias_initializer": {"class_name": "Zeros", "config": {}}, "kernel_regularizer": null, "bias_regularizer": null, "activity_regularizer": null, "kernel_constraint": null, "bias_constraint": null}}]}, "keras_version": "2.4.0", "backend": "tensorflow"}, "config_type": "keras" } }, "role": { "host": { "0": { "reader_0": { "table": { "name": "homo_host_mnist_train", "namespace": "experiment" } }, "dataio_0": { "with_label": true } } }, "guest": { "0": { "reader_0": { "table": { "name": "homo_guest_mnist_train", "namespace": "experiment" } }, "dataio_0": { "with_label": true, "output_format": "dense" } } } } } }