hetero_nn_testsuite.json 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. {
  2. "data": [
  3. {
  4. "file": "examples/data/breast_hetero_guest.csv",
  5. "head": 1,
  6. "partition": 16,
  7. "table_name": "breast_hetero_guest",
  8. "namespace": "experiment",
  9. "role": "guest_0"
  10. },
  11. {
  12. "file": "examples/data/breast_hetero_host.csv",
  13. "head": 1,
  14. "partition": 16,
  15. "table_name": "breast_hetero_host",
  16. "namespace": "experiment",
  17. "role": "host_0"
  18. },
  19. {
  20. "file": "examples/data/breast_hetero_host.csv",
  21. "head": 1,
  22. "partition": 16,
  23. "table_name": "breast_hetero_host",
  24. "namespace": "experiment",
  25. "role": "host_1"
  26. },
  27. {
  28. "file": "examples/data/vehicle_scale_hetero_guest.csv",
  29. "head": 1,
  30. "partition": 16,
  31. "table_name": "vehicle_scale_hetero_guest",
  32. "namespace": "experiment",
  33. "role": "guest_0"
  34. },
  35. {
  36. "file": "examples/data/vehicle_scale_hetero_host.csv",
  37. "head": 1,
  38. "partition": 16,
  39. "table_name": "vehicle_scale_hetero_host",
  40. "namespace": "experiment",
  41. "role": "host_0"
  42. }
  43. ],
  44. "pipeline_tasks": {
  45. "train_binary": {
  46. "script": "./pipeline-hetero-nn-train-binary.py"
  47. },
  48. "train_binary_with_coae_label_confuser": {
  49. "script": "./pipeline-hetero-nn-train-binary-coae.py"
  50. },
  51. "train_binary_multi_host": {
  52. "script": "./pipeline-hetero-nn-train-binary-multi-host.py"
  53. },
  54. "train_multi": {
  55. "script": "./pipeline-hetero-nn-train-multi.py"
  56. },
  57. "train_with_early_stop": {
  58. "script": "./pipeline-hetero-nn-train-with-early-stop.py"
  59. },
  60. "train_with_drop_out": {
  61. "script": "./pipeline-hetero-nn-train-binary-drop-out.py"
  62. },
  63. "train_with_selective-bp": {
  64. "script": "./pipeline-hetero-nn-train-binary-selective-bp.py"
  65. },
  66. "hetero_nn_train_with_warm_start": {
  67. "script": "./pipeline-hetero-nn-train-with-warm_start.py"
  68. },
  69. "hetero_nn_train_with_check_point": {
  70. "script": "./pipeline-hetero-nn-train-with-check-point.py"
  71. }
  72. }
  73. }