pems_bay.yaml 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. # SPDX-License-Identifier: Apache-2.0
  2. _target_: data.datasets.create_datasets
  3. config:
  4. source_path: /workspace/datasets/pems_bay/pems_bay.csv
  5. dest_path: /workspace/datasets/pems_bay/
  6. binarized: False
  7. graph: graph.bin
  8. graph_partitions: 1
  9. partition_joining_coef: 1
  10. train_range:
  11. - '2017-1-1'
  12. - '2017-5-8'
  13. valid_range:
  14. - '2017-5-8'
  15. - [2017,5,25,17,50]
  16. test_range:
  17. - [2017,5,25,17,50]
  18. - '2017-7-01'
  19. dataset_stride: 1
  20. scale_per_id: True
  21. encoder_length: 12
  22. input_length: 12
  23. example_length: 24
  24. MultiID: True
  25. features:
  26. - name: 'id'
  27. feature_type: 'ID'
  28. feature_embed_type: 'CATEGORICAL'
  29. cardinality: 325
  30. - name: 'Timestamp'
  31. feature_type: 'TIME'
  32. feature_embed_type: 'DATE'
  33. - name: 'Avg Speed'
  34. feature_type: 'TARGET'
  35. feature_embed_type: 'CONTINUOUS'
  36. scaler:
  37. _target_: sklearn.preprocessing.StandardScaler
  38. - name: 'Station'
  39. feature_type: 'STATIC'
  40. feature_embed_type: 'CATEGORICAL'
  41. cardinality: 326
  42. - name: 'Freeway #'
  43. feature_type: 'STATIC'
  44. feature_embed_type: 'CATEGORICAL'
  45. cardinality: 9
  46. - name: 'Direction of Travel'
  47. feature_type: 'STATIC'
  48. feature_embed_type: 'CATEGORICAL'
  49. cardinality: 5
  50. #- name: 'Station Length'
  51. # feature_type: 'STATIC'
  52. # feature_embed_type: 'CONTINUOUS'
  53. # scaler:
  54. # _target_: sklearn.preprocessing.StandardScaler
  55. - name: 'Avg Occupancy'
  56. feature_type: 'OBSERVED'
  57. feature_embed_type: 'CONTINUOUS'
  58. scaler:
  59. _target_: sklearn.preprocessing.StandardScaler
  60. - name: 'Total Flow'
  61. feature_type: 'OBSERVED'
  62. feature_embed_type: 'CONTINUOUS'
  63. scaler:
  64. _target_: sklearn.preprocessing.StandardScaler
  65. - name: 'Day of week'
  66. feature_type: 'KNOWN'
  67. feature_embed_type: 'CATEGORICAL'
  68. cardinality: 8
  69. - name: 'Month'
  70. feature_type: 'KNOWN'
  71. feature_embed_type: 'CONTINUOUS'
  72. scaler:
  73. _target_: sklearn.preprocessing.StandardScaler
  74. - name: 'Day'
  75. feature_type: 'KNOWN'
  76. feature_embed_type: 'CONTINUOUS'
  77. scaler:
  78. _target_: sklearn.preprocessing.StandardScaler
  79. - name: 'Hour'
  80. feature_type: 'KNOWN'
  81. feature_embed_type: 'CONTINUOUS'
  82. scaler:
  83. _target_: sklearn.preprocessing.StandardScaler
  84. - name: 'Minute'
  85. feature_type: 'KNOWN'
  86. feature_embed_type: 'CONTINUOUS'
  87. scaler:
  88. _target_: sklearn.preprocessing.StandardScaler
  89. time_series_count: 325