# Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#           http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

_target_: data.datasets.create_datasets
config:
    source_path: /workspace/datasets/traffic/traffic.csv
    dest_path: /workspace/datasets/traffic/
    time_ids: 'sensor_day'
    train_range:
      - 0
      - 151
    valid_range:
      - 144
      - 166
    test_range:
      - 159
      - 2000
    dataset_stride: 1
    scale_per_id: False
    encoder_length: 168
    example_length: 192
    MultiID: False
    features:
      - name: 'id'
        feature_type: 'ID'
        feature_embed_type: 'CATEGORICAL'
        cardinality: 964
      - name: 'hours_from_start'
        feature_type: 'TIME'
        feature_embed_type: 'CONTINUOUS'
      - name: 'values'
        feature_type: 'TARGET'
        feature_embed_type: 'CONTINUOUS'
        scaler:
            _target_: sklearn.preprocessing.StandardScaler
      - name: 'time_on_day'
        feature_type: 'KNOWN'
        feature_embed_type: 'CONTINUOUS'
        scaler:
            _target_: sklearn.preprocessing.StandardScaler
      - name: 'day_of_week'
        feature_type: 'KNOWN'
        feature_embed_type: 'CATEGORICAL'
        cardinality: 8
      - name: 'hours_from_start'
        feature_type: 'KNOWN'
        feature_embed_type: 'CONTINUOUS'
        scaler:
            _target_: sklearn.preprocessing.StandardScaler
      - name: 'categorical_id'
        feature_type: 'STATIC'
        feature_embed_type: 'CATEGORICAL'
        cardinality: 964
    train_samples: 450000
    valid_samples: 50000
    binarized: True
    time_series_count: 964
