M5_xgb.yaml 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. # SPDX-License-Identifier: MIT
  2. _target_: data.datasets.create_datasets
  3. config:
  4. source_path: /workspace/datasets/M5/M5.csv
  5. dest_path: /workspace/datasets/M5/
  6. iterable: False
  7. encoder_length: 28
  8. input_length: 28
  9. example_length: 56
  10. valid_boundary: '2016-04-25'
  11. train_samples: 1000000
  12. time_series_count: 30490
  13. drop_unseen: True
  14. MultiID: False
  15. features:
  16. - name: 'id'
  17. feature_type: 'ID'
  18. feature_embed_type: 'CATEGORICAL'
  19. cardinality: 30490
  20. - name: "date"
  21. feature_type: 'TIME'
  22. feature_embed_type: 'DATE'
  23. - name: "weight"
  24. feature_type: 'WEIGHT'
  25. feature_embed_type: 'CONTINUOUS'
  26. - name: "item_id"
  27. feature_type: 'STATIC'
  28. feature_embed_type: 'CATEGORICAL'
  29. cardinality: 3050
  30. - name: "dept_id"
  31. feature_type: 'STATIC'
  32. feature_embed_type: 'CATEGORICAL'
  33. cardinality: 8
  34. - name: "cat_id"
  35. feature_type: 'STATIC'
  36. feature_embed_type: 'CATEGORICAL'
  37. cardinality: 4
  38. - name: "store_id"
  39. feature_type: 'STATIC'
  40. feature_embed_type: 'CATEGORICAL'
  41. cardinality: 11
  42. - name: "state_id"
  43. feature_type: 'STATIC'
  44. feature_embed_type: 'CATEGORICAL'
  45. cardinality: 4
  46. - name: "items_sold"
  47. feature_type: 'TARGET'
  48. feature_embed_type: 'CONTINUOUS'
  49. scaler:
  50. _target_: data.data_utils.Log1pScaler
  51. - name: "wday"
  52. feature_type: 'KNOWN'
  53. feature_embed_type: 'CATEGORICAL'
  54. cardinality: 8
  55. - name: "month"
  56. feature_type: 'KNOWN'
  57. feature_embed_type: 'CATEGORICAL'
  58. cardinality: 13
  59. - name: "event_name_1"
  60. feature_type: 'KNOWN'
  61. feature_embed_type: 'CATEGORICAL'
  62. cardinality: 31
  63. - name: "event_type_1"
  64. feature_type: 'KNOWN'
  65. feature_embed_type: 'CATEGORICAL'
  66. cardinality: 5
  67. - name: "event_type_2"
  68. feature_type: 'KNOWN'
  69. feature_embed_type: 'CATEGORICAL'
  70. cardinality: 3
  71. - name: "event_name_2"
  72. feature_type: 'KNOWN'
  73. feature_embed_type: 'CATEGORICAL'
  74. cardinality: 5
  75. - name: "snap_CA"
  76. feature_type: 'KNOWN'
  77. feature_embed_type: 'CATEGORICAL'
  78. cardinality: 3
  79. - name: "snap_TX"
  80. feature_type: 'KNOWN'
  81. feature_embed_type: 'CATEGORICAL'
  82. cardinality: 3
  83. - name: "snap_WI"
  84. feature_type: 'KNOWN'
  85. feature_embed_type: 'CATEGORICAL'
  86. cardinality: 3
  87. - name: "sell_price"
  88. feature_type: 'KNOWN'
  89. feature_embed_type: 'CONTINUOUS'
  90. scaler:
  91. _target_: sklearn.preprocessing.StandardScaler
  92. binarized: False