hetero_nn_breast_config.yaml 428 B

12345678910111213141516171819
  1. data_guest: "examples/data/breast_hetero_guest.csv"
  2. data_host: "examples/data/breast_hetero_host.csv"
  3. idx: "id"
  4. label_name: "y"
  5. loss: "binary_crossentropy"
  6. learning_rate: 0.15
  7. tol: 0.01
  8. batch_size: 320
  9. metrics: ["AUC"]
  10. epochs: 100
  11. bottom_layer_units: 2
  12. interactive_layer_units: 2
  13. top_layer_units: 1
  14. guest_table_name: "breast_hetero_guest"
  15. host_table_name: "breast_hetero_host"
  16. top_act: "sigmoid"
  17. opt: "SGD"
  18. shape1: 10
  19. shape2: 20