informer_model/configs/experiments/informer-btcusdt-30m-quantile.yaml
Filip Stefaniuk b7c1470b08 update config
2024-10-17 21:23:30 +02:00

51 lines
1006 B
YAML

future_window:
value: 5
past_window:
value: 72
batch_size:
value: 128
max_epochs:
value: 50
data:
value:
dataset: "btc-usdt-30m:latest"
sliding_window: 0
validation: 0.2
fields:
value:
time_index: "time_index"
target: "returns"
group_ids: ["group_id"]
dynamic_unknown_real:
- "high_price"
- "low_price"
- "open_price"
- "close_price"
- "volume"
- "returns"
dynamic_unknown_cat: []
dynamic_known_real:
- "effective_rates"
- "vix_close_price"
- "fear_greed_index"
dynamic_known_cat:
- "hour"
- "weekday"
static_real: []
static_cat: []
loss:
value:
name: "Quantile"
quantiles: [0.01, 0.02, 0.03, 0.05, 0.1, 0.25, 0.5, 0.75, 0.9, 0.95, 0.97, 0.98, 0.99]
model:
value:
name: "Informer"
d_model: 1024
d_fully_connected: 1024
n_attention_heads: 2
dropout: 0.05
n_encoder_layers: 1
n_decoder_layers: 3
learning_rate: 0.0001
optimizer: "Adam"