-
Notifications
You must be signed in to change notification settings - Fork 13
Expand file tree
/
Copy pathconfig.yaml
More file actions
26 lines (26 loc) · 842 Bytes
/
Copy pathconfig.yaml
File metadata and controls
26 lines (26 loc) · 842 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
common:
task_type: generic_timeseries_forecasting
target_device: F28P55
dataset:
dataset_name: generic_timeseries_forecasting
input_data_path: https://software-dl.ti.com/C2000/esd/mcu_ai/datasets/generic_timeseries_forecasting.zip
data_processing_feature_extraction:
data_proc_transforms:
- SimpleWindow # 'SimpleWindow' must be specified for forecasting tasks
frame_size: 32
stride_size: 0.1
forecast_horizon: 2 # Number of future timesteps to be predicted
variables: 1 # Only temperature column (after timestamp)
target_variables: # Predict temperature
- 0 # Column index 0 = temperature (first column after timestamp)
training:
model_name: FCST_LSTM10
model_config: ''
batch_size: 32
training_epochs: 50
num_gpus: 1
quantization: 2
optimizer: adam
output_int: false
testing: {}
compilation: {}