forked from PaddlePaddle/PaddleX
-
Notifications
You must be signed in to change notification settings - Fork 0
/
DLinear.yaml
38 lines (33 loc) · 802 Bytes
/
DLinear.yaml
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
27
28
29
30
31
32
33
34
35
36
37
38
Global:
model: DLinear
mode: check_dataset # check_dataset/train/evaluate/predict
dataset_dir: "/paddle/dataset/paddlex/ts_fc/ts_dataset_examples"
device: gpu:0
output: "output"
CheckDataset:
convert:
enable: False
src_dataset_type: null
split:
enable: False
train_percent: null
val_percent: null
Train:
time_col: date
target_cols: OT
freq: 1h
input_len: 96
predict_len: 96
epochs_iters: 40
batch_size: 16
learning_rate: 0.0001
patience: 10
log_interval: 10
Evaluate:
weight_path: "output/best_model/model.pdparams"
Export:
weight_path: https://paddlets.bj.bcebos.com/dygraph/DLinear.pdparams.tar
Predict:
batch_size: 1
model_dir: "output/inference"
input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/ts/demo_ts/ts_fc.csv"