Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- experiments/tedit/synth-m/seed0/config.template.yaml +160 -0
- experiments/tedit/synth-m/seed0/results/expected_seed_metrics.json +21 -0
- experiments/tedit/synth-m/seed0/summary.json +5 -0
- experiments/tedit/synth-m/seed1/config.template.yaml +160 -0
- experiments/tedit/synth-m/seed1/results/expected_seed_metrics.json +21 -0
- experiments/tedit/synth-m/seed1/summary.json +5 -0
- experiments/tedit/synth-m/seed2/config.template.yaml +160 -0
- experiments/tedit/synth-m/seed2/results/expected_seed_metrics.json +21 -0
- experiments/tedit/synth-m/seed2/summary.json +5 -0
- experiments/tedit/synth-u/seed0/config.template.yaml +159 -0
- experiments/tedit/synth-u/seed0/results/expected_seed_metrics.json +21 -0
- experiments/tedit/synth-u/seed0/summary.json +5 -0
- experiments/tedit/synth-u/seed1/config.template.yaml +159 -0
- experiments/tedit/synth-u/seed1/results/expected_seed_metrics.json +21 -0
- experiments/tedit/synth-u/seed1/summary.json +5 -0
- experiments/tedit/synth-u/seed2/config.template.yaml +159 -0
- experiments/tedit/synth-u/seed2/results/expected_seed_metrics.json +21 -0
- experiments/tedit/synth-u/seed2/summary.json +5 -0
- experiments/timevqvae/synth-m/seed0/config.template.yaml +182 -0
- experiments/timevqvae/synth-m/seed0/results/expected_seed_metrics.json +21 -0
- experiments/timevqvae/synth-m/seed0/summary.json +5 -0
- experiments/timevqvae/synth-m/seed1/config.template.yaml +182 -0
- experiments/timevqvae/synth-m/seed1/results/expected_seed_metrics.json +21 -0
- experiments/timevqvae/synth-m/seed1/summary.json +5 -0
- experiments/timevqvae/synth-m/seed2/config.template.yaml +182 -0
- experiments/timevqvae/synth-m/seed2/results/expected_seed_metrics.json +21 -0
- experiments/timevqvae/synth-m/seed2/summary.json +5 -0
- experiments/timevqvae/synth-u/seed2/results/expected_seed_metrics.json +21 -0
- experiments/timeweaver/synth-m/seed0/config.template.yaml +155 -0
- experiments/timeweaver/synth-m/seed0/summary.json +5 -0
- experiments/timeweaver/synth-u/seed0/config.template.yaml +154 -0
- experiments/timeweaver/synth-u/seed0/results/expected_seed_metrics.json +21 -0
- experiments/timeweaver/synth-u/seed0/summary.json +5 -0
- experiments/timeweaver/synth-u/seed1/config.template.yaml +154 -0
- experiments/timeweaver/synth-u/seed1/results/expected_seed_metrics.json +21 -0
- experiments/timeweaver/synth-u/seed1/summary.json +5 -0
- experiments/timeweaver/synth-u/seed2/config.template.yaml +154 -0
- experiments/timeweaver/synth-u/seed2/results/expected_seed_metrics.json +21 -0
- experiments/timeweaver/synth-u/seed2/summary.json +5 -0
- experiments/verbalts/synth-m/seed0/config.template.yaml +151 -0
- experiments/verbalts/synth-m/seed0/results/expected_seed_metrics.json +21 -0
- experiments/verbalts/synth-m/seed0/summary.json +5 -0
- experiments/verbalts/synth-m/seed1/config.template.yaml +151 -0
- experiments/verbalts/synth-m/seed1/results/expected_seed_metrics.json +21 -0
- experiments/verbalts/synth-m/seed1/summary.json +5 -0
- experiments/verbalts/synth-m/seed2/config.template.yaml +151 -0
- experiments/verbalts/synth-m/seed2/results/expected_seed_metrics.json +21 -0
- experiments/verbalts/synth-m/seed2/summary.json +5 -0
- experiments/verbalts/synth-u/seed0/config.template.yaml +151 -0
- experiments/verbalts/synth-u/seed0/results/expected_seed_metrics.json +21 -0
experiments/tedit/synth-m/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,160 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_tedit
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: tedit
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
multipatch_num: 3
|
| 47 |
+
base_patch: 4
|
| 48 |
+
L_patch_len: 3
|
| 49 |
+
diffusion_steps: 50
|
| 50 |
+
diffusion_embedding_dim: 128
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
is_linear: false
|
| 56 |
+
side_var_emb: 16
|
| 57 |
+
side_time_emb: 112
|
| 58 |
+
var_emb: 64
|
| 59 |
+
time_emb: 64
|
| 60 |
+
attr_proj_type: avg
|
| 61 |
+
attr_heads: 8
|
| 62 |
+
attr_layers: 2
|
| 63 |
+
n_attrs: 4
|
| 64 |
+
feature_ratio: 0.5
|
| 65 |
+
condition:
|
| 66 |
+
text:
|
| 67 |
+
enabled: false
|
| 68 |
+
input_dim: 1024
|
| 69 |
+
embedding_key: cap_emb
|
| 70 |
+
dropout: 0.0
|
| 71 |
+
cfg_scale: 1.0
|
| 72 |
+
attribute:
|
| 73 |
+
enabled: true
|
| 74 |
+
continuous_dim: 0
|
| 75 |
+
discrete_configs:
|
| 76 |
+
- num_classes: 4
|
| 77 |
+
- num_classes: 2
|
| 78 |
+
- num_classes: 4
|
| 79 |
+
- num_classes: 4
|
| 80 |
+
output_dim: 128
|
| 81 |
+
dropout: 0.0
|
| 82 |
+
label:
|
| 83 |
+
enabled: false
|
| 84 |
+
num_classes: 10
|
| 85 |
+
output_dim: 64
|
| 86 |
+
fusion: concat
|
| 87 |
+
condition_dropout: 0.0
|
| 88 |
+
eval:
|
| 89 |
+
n_samples: 10
|
| 90 |
+
metrics:
|
| 91 |
+
- dtw
|
| 92 |
+
- wape
|
| 93 |
+
- ed
|
| 94 |
+
- crps
|
| 95 |
+
- acd
|
| 96 |
+
- sd
|
| 97 |
+
- kd
|
| 98 |
+
- mdd
|
| 99 |
+
- fid
|
| 100 |
+
- sfd
|
| 101 |
+
- jftsd
|
| 102 |
+
- prdc_f1
|
| 103 |
+
- joint_prdc_f1
|
| 104 |
+
- cttp
|
| 105 |
+
- disc_auc
|
| 106 |
+
batch_size: 128
|
| 107 |
+
save_samples: true
|
| 108 |
+
sampler: ddim
|
| 109 |
+
display_interval: 10
|
| 110 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 111 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 112 |
+
cache_folder: __CACHE_ROOT__/tedit/synth-m/seed0
|
| 113 |
+
use_longalign: false
|
| 114 |
+
reference_split: train
|
| 115 |
+
disc_auc:
|
| 116 |
+
k_folds: 5
|
| 117 |
+
epochs: 5
|
| 118 |
+
batch_size: 128
|
| 119 |
+
lr: 0.001
|
| 120 |
+
weight_decay: 0.0001
|
| 121 |
+
hidden_channels: 128
|
| 122 |
+
num_layers: 3
|
| 123 |
+
dropout: 0.1
|
| 124 |
+
seed: 42
|
| 125 |
+
num_workers: 0
|
| 126 |
+
prdc:
|
| 127 |
+
k: 5
|
| 128 |
+
max_samples: 5000
|
| 129 |
+
seed: 0
|
| 130 |
+
metric: euclidean
|
| 131 |
+
backend: auto
|
| 132 |
+
knn_chunk_size: 4096
|
| 133 |
+
joint_enable: true
|
| 134 |
+
joint_weights:
|
| 135 |
+
ts: 1.0
|
| 136 |
+
text: 1.0
|
| 137 |
+
joint_normalize: standard
|
| 138 |
+
segment_classifier:
|
| 139 |
+
enable: false
|
| 140 |
+
segment_len: 128
|
| 141 |
+
n_segments: 3
|
| 142 |
+
seed: 42
|
| 143 |
+
acd_max_lag: 50
|
| 144 |
+
mdd_bins: 32
|
| 145 |
+
use_cache: false
|
| 146 |
+
cache_file: predictions_cache.pkl
|
| 147 |
+
viz:
|
| 148 |
+
enable: true
|
| 149 |
+
k_cases: 20
|
| 150 |
+
max_vars: 8
|
| 151 |
+
ncols: 5
|
| 152 |
+
seed: 42
|
| 153 |
+
dpi: 200
|
| 154 |
+
alpha: 0.25
|
| 155 |
+
figsize_per_subplot:
|
| 156 |
+
- 3.0
|
| 157 |
+
- 2.0
|
| 158 |
+
output_type: pdf
|
| 159 |
+
output_dir: ''
|
| 160 |
+
created_at: '2025-12-29T11:59:19.476747'
|
experiments/tedit/synth-m/seed0/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.449965004665586,
|
| 3 |
+
"wape": 84.70754473291974,
|
| 4 |
+
"ed": 12.568117126464843,
|
| 5 |
+
"crps": 0.43670245952221176,
|
| 6 |
+
"acd": 0.05856379494071007,
|
| 7 |
+
"sd": 0.029261844309823173,
|
| 8 |
+
"kd": 0.38030240728596487,
|
| 9 |
+
"mdd": 0.01425675455729166,
|
| 10 |
+
"fid": 34.12589442222904,
|
| 11 |
+
"sfd": 29.040304538337182,
|
| 12 |
+
"jftsd": 42.52122642823781,
|
| 13 |
+
"prdc_f1.f1": 0.3770247463499134,
|
| 14 |
+
"prdc_f1.precision": 0.7595,
|
| 15 |
+
"prdc_f1.recall": 0.25075,
|
| 16 |
+
"joint_prdc_f1.f1": 0.49005342399222923,
|
| 17 |
+
"joint_prdc_f1.precision": 0.6275,
|
| 18 |
+
"joint_prdc_f1.recall": 0.402,
|
| 19 |
+
"cttp": 21.539698181152342,
|
| 20 |
+
"disc_auc": 0.99049265625
|
| 21 |
+
}
|
experiments/tedit/synth-m/seed0/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T12:45:10.406925",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/tedit/synth-m/seed1/config.template.yaml
ADDED
|
@@ -0,0 +1,160 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_tedit
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 1
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: tedit
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
multipatch_num: 3
|
| 47 |
+
base_patch: 4
|
| 48 |
+
L_patch_len: 3
|
| 49 |
+
diffusion_steps: 50
|
| 50 |
+
diffusion_embedding_dim: 128
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
is_linear: false
|
| 56 |
+
side_var_emb: 16
|
| 57 |
+
side_time_emb: 112
|
| 58 |
+
var_emb: 64
|
| 59 |
+
time_emb: 64
|
| 60 |
+
attr_proj_type: avg
|
| 61 |
+
attr_heads: 8
|
| 62 |
+
attr_layers: 2
|
| 63 |
+
n_attrs: 4
|
| 64 |
+
feature_ratio: 0.5
|
| 65 |
+
condition:
|
| 66 |
+
text:
|
| 67 |
+
enabled: false
|
| 68 |
+
input_dim: 1024
|
| 69 |
+
embedding_key: cap_emb
|
| 70 |
+
dropout: 0.0
|
| 71 |
+
cfg_scale: 1.0
|
| 72 |
+
attribute:
|
| 73 |
+
enabled: true
|
| 74 |
+
continuous_dim: 0
|
| 75 |
+
discrete_configs:
|
| 76 |
+
- num_classes: 4
|
| 77 |
+
- num_classes: 2
|
| 78 |
+
- num_classes: 4
|
| 79 |
+
- num_classes: 4
|
| 80 |
+
output_dim: 128
|
| 81 |
+
dropout: 0.0
|
| 82 |
+
label:
|
| 83 |
+
enabled: false
|
| 84 |
+
num_classes: 10
|
| 85 |
+
output_dim: 64
|
| 86 |
+
fusion: concat
|
| 87 |
+
condition_dropout: 0.0
|
| 88 |
+
eval:
|
| 89 |
+
n_samples: 10
|
| 90 |
+
metrics:
|
| 91 |
+
- dtw
|
| 92 |
+
- wape
|
| 93 |
+
- ed
|
| 94 |
+
- crps
|
| 95 |
+
- acd
|
| 96 |
+
- sd
|
| 97 |
+
- kd
|
| 98 |
+
- mdd
|
| 99 |
+
- fid
|
| 100 |
+
- sfd
|
| 101 |
+
- jftsd
|
| 102 |
+
- prdc_f1
|
| 103 |
+
- joint_prdc_f1
|
| 104 |
+
- cttp
|
| 105 |
+
- disc_auc
|
| 106 |
+
batch_size: 128
|
| 107 |
+
save_samples: true
|
| 108 |
+
sampler: ddim
|
| 109 |
+
display_interval: 10
|
| 110 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 111 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 112 |
+
cache_folder: __CACHE_ROOT__/tedit/synth-m/seed1
|
| 113 |
+
use_longalign: false
|
| 114 |
+
reference_split: train
|
| 115 |
+
disc_auc:
|
| 116 |
+
k_folds: 5
|
| 117 |
+
epochs: 5
|
| 118 |
+
batch_size: 128
|
| 119 |
+
lr: 0.001
|
| 120 |
+
weight_decay: 0.0001
|
| 121 |
+
hidden_channels: 128
|
| 122 |
+
num_layers: 3
|
| 123 |
+
dropout: 0.1
|
| 124 |
+
seed: 42
|
| 125 |
+
num_workers: 0
|
| 126 |
+
prdc:
|
| 127 |
+
k: 5
|
| 128 |
+
max_samples: 5000
|
| 129 |
+
seed: 0
|
| 130 |
+
metric: euclidean
|
| 131 |
+
backend: auto
|
| 132 |
+
knn_chunk_size: 4096
|
| 133 |
+
joint_enable: true
|
| 134 |
+
joint_weights:
|
| 135 |
+
ts: 1.0
|
| 136 |
+
text: 1.0
|
| 137 |
+
joint_normalize: standard
|
| 138 |
+
segment_classifier:
|
| 139 |
+
enable: false
|
| 140 |
+
segment_len: 128
|
| 141 |
+
n_segments: 3
|
| 142 |
+
seed: 42
|
| 143 |
+
acd_max_lag: 50
|
| 144 |
+
mdd_bins: 32
|
| 145 |
+
use_cache: false
|
| 146 |
+
cache_file: predictions_cache.pkl
|
| 147 |
+
viz:
|
| 148 |
+
enable: true
|
| 149 |
+
k_cases: 20
|
| 150 |
+
max_vars: 8
|
| 151 |
+
ncols: 5
|
| 152 |
+
seed: 42
|
| 153 |
+
dpi: 200
|
| 154 |
+
alpha: 0.25
|
| 155 |
+
figsize_per_subplot:
|
| 156 |
+
- 3.0
|
| 157 |
+
- 2.0
|
| 158 |
+
output_type: pdf
|
| 159 |
+
output_dir: ''
|
| 160 |
+
created_at: '2025-12-29T12:00:28.391672'
|
experiments/tedit/synth-m/seed1/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 10.657325486540197,
|
| 3 |
+
"wape": 93.90555822286085,
|
| 4 |
+
"ed": 13.747008003234864,
|
| 5 |
+
"crps": 0.48623683406631973,
|
| 6 |
+
"acd": 0.04866933822631836,
|
| 7 |
+
"sd": 0.02613117771252848,
|
| 8 |
+
"kd": 0.18543531249497258,
|
| 9 |
+
"mdd": 0.017248840332031244,
|
| 10 |
+
"fid": 35.71198160860902,
|
| 11 |
+
"sfd": 28.30212871877896,
|
| 12 |
+
"jftsd": 44.363306631870586,
|
| 13 |
+
"prdc_f1.f1": 0.32694431216931213,
|
| 14 |
+
"prdc_f1.precision": 0.73475,
|
| 15 |
+
"prdc_f1.recall": 0.21025,
|
| 16 |
+
"joint_prdc_f1.f1": 0.47389876605389064,
|
| 17 |
+
"joint_prdc_f1.precision": 0.602,
|
| 18 |
+
"joint_prdc_f1.recall": 0.39075,
|
| 19 |
+
"cttp": 20.862328704833985,
|
| 20 |
+
"disc_auc": 0.9933971875000001
|
| 21 |
+
}
|
experiments/tedit/synth-m/seed1/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T12:35:21.171409",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/tedit/synth-m/seed2/config.template.yaml
ADDED
|
@@ -0,0 +1,160 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_tedit
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 2
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: tedit
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
multipatch_num: 3
|
| 47 |
+
base_patch: 4
|
| 48 |
+
L_patch_len: 3
|
| 49 |
+
diffusion_steps: 50
|
| 50 |
+
diffusion_embedding_dim: 128
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
is_linear: false
|
| 56 |
+
side_var_emb: 16
|
| 57 |
+
side_time_emb: 112
|
| 58 |
+
var_emb: 64
|
| 59 |
+
time_emb: 64
|
| 60 |
+
attr_proj_type: avg
|
| 61 |
+
attr_heads: 8
|
| 62 |
+
attr_layers: 2
|
| 63 |
+
n_attrs: 4
|
| 64 |
+
feature_ratio: 0.5
|
| 65 |
+
condition:
|
| 66 |
+
text:
|
| 67 |
+
enabled: false
|
| 68 |
+
input_dim: 1024
|
| 69 |
+
embedding_key: cap_emb
|
| 70 |
+
dropout: 0.0
|
| 71 |
+
cfg_scale: 1.0
|
| 72 |
+
attribute:
|
| 73 |
+
enabled: true
|
| 74 |
+
continuous_dim: 0
|
| 75 |
+
discrete_configs:
|
| 76 |
+
- num_classes: 4
|
| 77 |
+
- num_classes: 2
|
| 78 |
+
- num_classes: 4
|
| 79 |
+
- num_classes: 4
|
| 80 |
+
output_dim: 128
|
| 81 |
+
dropout: 0.0
|
| 82 |
+
label:
|
| 83 |
+
enabled: false
|
| 84 |
+
num_classes: 10
|
| 85 |
+
output_dim: 64
|
| 86 |
+
fusion: concat
|
| 87 |
+
condition_dropout: 0.0
|
| 88 |
+
eval:
|
| 89 |
+
n_samples: 10
|
| 90 |
+
metrics:
|
| 91 |
+
- dtw
|
| 92 |
+
- wape
|
| 93 |
+
- ed
|
| 94 |
+
- crps
|
| 95 |
+
- acd
|
| 96 |
+
- sd
|
| 97 |
+
- kd
|
| 98 |
+
- mdd
|
| 99 |
+
- fid
|
| 100 |
+
- sfd
|
| 101 |
+
- jftsd
|
| 102 |
+
- prdc_f1
|
| 103 |
+
- joint_prdc_f1
|
| 104 |
+
- cttp
|
| 105 |
+
- disc_auc
|
| 106 |
+
batch_size: 128
|
| 107 |
+
save_samples: true
|
| 108 |
+
sampler: ddim
|
| 109 |
+
display_interval: 10
|
| 110 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 111 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 112 |
+
cache_folder: __CACHE_ROOT__/tedit/synth-m/seed2
|
| 113 |
+
use_longalign: false
|
| 114 |
+
reference_split: train
|
| 115 |
+
disc_auc:
|
| 116 |
+
k_folds: 5
|
| 117 |
+
epochs: 5
|
| 118 |
+
batch_size: 128
|
| 119 |
+
lr: 0.001
|
| 120 |
+
weight_decay: 0.0001
|
| 121 |
+
hidden_channels: 128
|
| 122 |
+
num_layers: 3
|
| 123 |
+
dropout: 0.1
|
| 124 |
+
seed: 42
|
| 125 |
+
num_workers: 0
|
| 126 |
+
prdc:
|
| 127 |
+
k: 5
|
| 128 |
+
max_samples: 5000
|
| 129 |
+
seed: 0
|
| 130 |
+
metric: euclidean
|
| 131 |
+
backend: auto
|
| 132 |
+
knn_chunk_size: 4096
|
| 133 |
+
joint_enable: true
|
| 134 |
+
joint_weights:
|
| 135 |
+
ts: 1.0
|
| 136 |
+
text: 1.0
|
| 137 |
+
joint_normalize: standard
|
| 138 |
+
segment_classifier:
|
| 139 |
+
enable: false
|
| 140 |
+
segment_len: 128
|
| 141 |
+
n_segments: 3
|
| 142 |
+
seed: 42
|
| 143 |
+
acd_max_lag: 50
|
| 144 |
+
mdd_bins: 32
|
| 145 |
+
use_cache: false
|
| 146 |
+
cache_file: predictions_cache.pkl
|
| 147 |
+
viz:
|
| 148 |
+
enable: true
|
| 149 |
+
k_cases: 20
|
| 150 |
+
max_vars: 8
|
| 151 |
+
ncols: 5
|
| 152 |
+
seed: 42
|
| 153 |
+
dpi: 200
|
| 154 |
+
alpha: 0.25
|
| 155 |
+
figsize_per_subplot:
|
| 156 |
+
- 3.0
|
| 157 |
+
- 2.0
|
| 158 |
+
output_type: pdf
|
| 159 |
+
output_dir: ''
|
| 160 |
+
created_at: '2025-12-29T12:08:28.447209'
|
experiments/tedit/synth-m/seed2/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.850579732121773,
|
| 3 |
+
"wape": 88.43444260454119,
|
| 4 |
+
"ed": 13.037744049072266,
|
| 5 |
+
"crps": 0.47628230624078943,
|
| 6 |
+
"acd": 0.05967891961336136,
|
| 7 |
+
"sd": 0.03868219061395673,
|
| 8 |
+
"kd": 0.1998955795200681,
|
| 9 |
+
"mdd": 0.014961975097656245,
|
| 10 |
+
"fid": 35.74539814735715,
|
| 11 |
+
"sfd": 28.039383014968706,
|
| 12 |
+
"jftsd": 44.55691533072158,
|
| 13 |
+
"prdc_f1.f1": 0.34751166448230664,
|
| 14 |
+
"prdc_f1.precision": 0.72525,
|
| 15 |
+
"prdc_f1.recall": 0.2285,
|
| 16 |
+
"joint_prdc_f1.f1": 0.47022108585858585,
|
| 17 |
+
"joint_prdc_f1.precision": 0.60575,
|
| 18 |
+
"joint_prdc_f1.recall": 0.38425,
|
| 19 |
+
"cttp": 21.313952178955077,
|
| 20 |
+
"disc_auc": 0.98900359375
|
| 21 |
+
}
|
experiments/tedit/synth-m/seed2/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T12:41:29.837867",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/tedit/synth-u/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_tedit
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: tedit
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
multipatch_num: 3
|
| 47 |
+
base_patch: 4
|
| 48 |
+
L_patch_len: 3
|
| 49 |
+
diffusion_steps: 50
|
| 50 |
+
diffusion_embedding_dim: 128
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
is_linear: false
|
| 56 |
+
side_var_emb: 16
|
| 57 |
+
side_time_emb: 112
|
| 58 |
+
var_emb: 64
|
| 59 |
+
time_emb: 64
|
| 60 |
+
attr_proj_type: avg
|
| 61 |
+
attr_heads: 8
|
| 62 |
+
attr_layers: 2
|
| 63 |
+
n_attrs: 4
|
| 64 |
+
feature_ratio: 0.5
|
| 65 |
+
condition:
|
| 66 |
+
text:
|
| 67 |
+
enabled: false
|
| 68 |
+
input_dim: 1024
|
| 69 |
+
embedding_key: cap_emb
|
| 70 |
+
dropout: 0.0
|
| 71 |
+
cfg_scale: 1.0
|
| 72 |
+
attribute:
|
| 73 |
+
enabled: true
|
| 74 |
+
continuous_dim: 0
|
| 75 |
+
discrete_configs:
|
| 76 |
+
- num_classes: 4
|
| 77 |
+
- num_classes: 2
|
| 78 |
+
- num_classes: 4
|
| 79 |
+
output_dim: 128
|
| 80 |
+
dropout: 0.0
|
| 81 |
+
label:
|
| 82 |
+
enabled: false
|
| 83 |
+
num_classes: 10
|
| 84 |
+
output_dim: 64
|
| 85 |
+
fusion: concat
|
| 86 |
+
condition_dropout: 0.0
|
| 87 |
+
eval:
|
| 88 |
+
n_samples: 10
|
| 89 |
+
metrics:
|
| 90 |
+
- dtw
|
| 91 |
+
- wape
|
| 92 |
+
- ed
|
| 93 |
+
- crps
|
| 94 |
+
- acd
|
| 95 |
+
- sd
|
| 96 |
+
- kd
|
| 97 |
+
- mdd
|
| 98 |
+
- fid
|
| 99 |
+
- sfd
|
| 100 |
+
- jftsd
|
| 101 |
+
- prdc_f1
|
| 102 |
+
- joint_prdc_f1
|
| 103 |
+
- cttp
|
| 104 |
+
- disc_auc
|
| 105 |
+
batch_size: 128
|
| 106 |
+
save_samples: true
|
| 107 |
+
sampler: ddim
|
| 108 |
+
display_interval: 10
|
| 109 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 110 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 111 |
+
cache_folder: __CACHE_ROOT__/tedit/synth-u/seed0
|
| 112 |
+
use_longalign: false
|
| 113 |
+
reference_split: train
|
| 114 |
+
disc_auc:
|
| 115 |
+
k_folds: 5
|
| 116 |
+
epochs: 5
|
| 117 |
+
batch_size: 128
|
| 118 |
+
lr: 0.001
|
| 119 |
+
weight_decay: 0.0001
|
| 120 |
+
hidden_channels: 128
|
| 121 |
+
num_layers: 3
|
| 122 |
+
dropout: 0.1
|
| 123 |
+
seed: 42
|
| 124 |
+
num_workers: 0
|
| 125 |
+
prdc:
|
| 126 |
+
k: 5
|
| 127 |
+
max_samples: 5000
|
| 128 |
+
seed: 0
|
| 129 |
+
metric: euclidean
|
| 130 |
+
backend: auto
|
| 131 |
+
knn_chunk_size: 4096
|
| 132 |
+
joint_enable: true
|
| 133 |
+
joint_weights:
|
| 134 |
+
ts: 1.0
|
| 135 |
+
text: 1.0
|
| 136 |
+
joint_normalize: standard
|
| 137 |
+
segment_classifier:
|
| 138 |
+
enable: false
|
| 139 |
+
segment_len: 128
|
| 140 |
+
n_segments: 3
|
| 141 |
+
seed: 42
|
| 142 |
+
acd_max_lag: 50
|
| 143 |
+
mdd_bins: 32
|
| 144 |
+
use_cache: false
|
| 145 |
+
cache_file: predictions_cache.pkl
|
| 146 |
+
viz:
|
| 147 |
+
enable: true
|
| 148 |
+
k_cases: 20
|
| 149 |
+
max_vars: 8
|
| 150 |
+
ncols: 5
|
| 151 |
+
seed: 42
|
| 152 |
+
dpi: 200
|
| 153 |
+
alpha: 0.25
|
| 154 |
+
figsize_per_subplot:
|
| 155 |
+
- 3.0
|
| 156 |
+
- 2.0
|
| 157 |
+
output_type: pdf
|
| 158 |
+
output_dir: ''
|
| 159 |
+
created_at: '2025-12-29T12:18:46.320258'
|
experiments/tedit/synth-u/seed0/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 7.959363302202195,
|
| 3 |
+
"wape": 125.00642335567366,
|
| 4 |
+
"ed": 12.238202255249023,
|
| 5 |
+
"crps": 0.7558069003698727,
|
| 6 |
+
"acd": 0.06724201887845993,
|
| 7 |
+
"sd": 0.28640310958163423,
|
| 8 |
+
"kd": 0.2388073990442381,
|
| 9 |
+
"mdd": 0.02708036295572916,
|
| 10 |
+
"fid": 53.56211119503274,
|
| 11 |
+
"sfd": 3.2625660021071496,
|
| 12 |
+
"jftsd": 64.02011808883037,
|
| 13 |
+
"prdc_f1.f1": 0.18506040784917277,
|
| 14 |
+
"prdc_f1.precision": 0.538,
|
| 15 |
+
"prdc_f1.recall": 0.11175,
|
| 16 |
+
"joint_prdc_f1.f1": 0.49510117967332123,
|
| 17 |
+
"joint_prdc_f1.precision": 0.7265,
|
| 18 |
+
"joint_prdc_f1.recall": 0.3755,
|
| 19 |
+
"cttp": 19.67137551879883,
|
| 20 |
+
"disc_auc": 0.98694375
|
| 21 |
+
}
|
experiments/tedit/synth-u/seed0/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T12:29:22.785295",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/tedit/synth-u/seed1/config.template.yaml
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_tedit
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 1
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: tedit
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
multipatch_num: 3
|
| 47 |
+
base_patch: 4
|
| 48 |
+
L_patch_len: 3
|
| 49 |
+
diffusion_steps: 50
|
| 50 |
+
diffusion_embedding_dim: 128
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
is_linear: false
|
| 56 |
+
side_var_emb: 16
|
| 57 |
+
side_time_emb: 112
|
| 58 |
+
var_emb: 64
|
| 59 |
+
time_emb: 64
|
| 60 |
+
attr_proj_type: avg
|
| 61 |
+
attr_heads: 8
|
| 62 |
+
attr_layers: 2
|
| 63 |
+
n_attrs: 4
|
| 64 |
+
feature_ratio: 0.5
|
| 65 |
+
condition:
|
| 66 |
+
text:
|
| 67 |
+
enabled: false
|
| 68 |
+
input_dim: 1024
|
| 69 |
+
embedding_key: cap_emb
|
| 70 |
+
dropout: 0.0
|
| 71 |
+
cfg_scale: 1.0
|
| 72 |
+
attribute:
|
| 73 |
+
enabled: true
|
| 74 |
+
continuous_dim: 0
|
| 75 |
+
discrete_configs:
|
| 76 |
+
- num_classes: 4
|
| 77 |
+
- num_classes: 2
|
| 78 |
+
- num_classes: 4
|
| 79 |
+
output_dim: 128
|
| 80 |
+
dropout: 0.0
|
| 81 |
+
label:
|
| 82 |
+
enabled: false
|
| 83 |
+
num_classes: 10
|
| 84 |
+
output_dim: 64
|
| 85 |
+
fusion: concat
|
| 86 |
+
condition_dropout: 0.0
|
| 87 |
+
eval:
|
| 88 |
+
n_samples: 10
|
| 89 |
+
metrics:
|
| 90 |
+
- dtw
|
| 91 |
+
- wape
|
| 92 |
+
- ed
|
| 93 |
+
- crps
|
| 94 |
+
- acd
|
| 95 |
+
- sd
|
| 96 |
+
- kd
|
| 97 |
+
- mdd
|
| 98 |
+
- fid
|
| 99 |
+
- sfd
|
| 100 |
+
- jftsd
|
| 101 |
+
- prdc_f1
|
| 102 |
+
- joint_prdc_f1
|
| 103 |
+
- cttp
|
| 104 |
+
- disc_auc
|
| 105 |
+
batch_size: 128
|
| 106 |
+
save_samples: true
|
| 107 |
+
sampler: ddim
|
| 108 |
+
display_interval: 10
|
| 109 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 110 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 111 |
+
cache_folder: __CACHE_ROOT__/tedit/synth-u/seed1
|
| 112 |
+
use_longalign: false
|
| 113 |
+
reference_split: train
|
| 114 |
+
disc_auc:
|
| 115 |
+
k_folds: 5
|
| 116 |
+
epochs: 5
|
| 117 |
+
batch_size: 128
|
| 118 |
+
lr: 0.001
|
| 119 |
+
weight_decay: 0.0001
|
| 120 |
+
hidden_channels: 128
|
| 121 |
+
num_layers: 3
|
| 122 |
+
dropout: 0.1
|
| 123 |
+
seed: 42
|
| 124 |
+
num_workers: 0
|
| 125 |
+
prdc:
|
| 126 |
+
k: 5
|
| 127 |
+
max_samples: 5000
|
| 128 |
+
seed: 0
|
| 129 |
+
metric: euclidean
|
| 130 |
+
backend: auto
|
| 131 |
+
knn_chunk_size: 4096
|
| 132 |
+
joint_enable: true
|
| 133 |
+
joint_weights:
|
| 134 |
+
ts: 1.0
|
| 135 |
+
text: 1.0
|
| 136 |
+
joint_normalize: standard
|
| 137 |
+
segment_classifier:
|
| 138 |
+
enable: false
|
| 139 |
+
segment_len: 128
|
| 140 |
+
n_segments: 3
|
| 141 |
+
seed: 42
|
| 142 |
+
acd_max_lag: 50
|
| 143 |
+
mdd_bins: 32
|
| 144 |
+
use_cache: false
|
| 145 |
+
cache_file: predictions_cache.pkl
|
| 146 |
+
viz:
|
| 147 |
+
enable: true
|
| 148 |
+
k_cases: 20
|
| 149 |
+
max_vars: 8
|
| 150 |
+
ncols: 5
|
| 151 |
+
seed: 42
|
| 152 |
+
dpi: 200
|
| 153 |
+
alpha: 0.25
|
| 154 |
+
figsize_per_subplot:
|
| 155 |
+
- 3.0
|
| 156 |
+
- 2.0
|
| 157 |
+
output_type: pdf
|
| 158 |
+
output_dir: ''
|
| 159 |
+
created_at: '2025-12-29T12:32:08.099443'
|
experiments/tedit/synth-u/seed1/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 6.888405716152939,
|
| 3 |
+
"wape": 108.86051449077287,
|
| 4 |
+
"ed": 10.816663734436036,
|
| 5 |
+
"crps": 0.648811074876578,
|
| 6 |
+
"acd": 0.06429418176412582,
|
| 7 |
+
"sd": 0.09525932871616731,
|
| 8 |
+
"kd": 0.41859134401745623,
|
| 9 |
+
"mdd": 0.021761311848958333,
|
| 10 |
+
"fid": 52.893325673487965,
|
| 11 |
+
"sfd": 2.6884680714630846,
|
| 12 |
+
"jftsd": 63.38460337243362,
|
| 13 |
+
"prdc_f1.f1": 0.1599864812668984,
|
| 14 |
+
"prdc_f1.precision": 0.55375,
|
| 15 |
+
"prdc_f1.recall": 0.0935,
|
| 16 |
+
"joint_prdc_f1.f1": 0.48190530389255626,
|
| 17 |
+
"joint_prdc_f1.precision": 0.74125,
|
| 18 |
+
"joint_prdc_f1.recall": 0.357,
|
| 19 |
+
"cttp": 20.302712860107423,
|
| 20 |
+
"disc_auc": 0.98518453125
|
| 21 |
+
}
|
experiments/tedit/synth-u/seed1/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T12:40:34.382593",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/tedit/synth-u/seed2/config.template.yaml
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_tedit
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 2
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: tedit
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
multipatch_num: 3
|
| 47 |
+
base_patch: 4
|
| 48 |
+
L_patch_len: 3
|
| 49 |
+
diffusion_steps: 50
|
| 50 |
+
diffusion_embedding_dim: 128
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
is_linear: false
|
| 56 |
+
side_var_emb: 16
|
| 57 |
+
side_time_emb: 112
|
| 58 |
+
var_emb: 64
|
| 59 |
+
time_emb: 64
|
| 60 |
+
attr_proj_type: avg
|
| 61 |
+
attr_heads: 8
|
| 62 |
+
attr_layers: 2
|
| 63 |
+
n_attrs: 4
|
| 64 |
+
feature_ratio: 0.5
|
| 65 |
+
condition:
|
| 66 |
+
text:
|
| 67 |
+
enabled: false
|
| 68 |
+
input_dim: 1024
|
| 69 |
+
embedding_key: cap_emb
|
| 70 |
+
dropout: 0.0
|
| 71 |
+
cfg_scale: 1.0
|
| 72 |
+
attribute:
|
| 73 |
+
enabled: true
|
| 74 |
+
continuous_dim: 0
|
| 75 |
+
discrete_configs:
|
| 76 |
+
- num_classes: 4
|
| 77 |
+
- num_classes: 2
|
| 78 |
+
- num_classes: 4
|
| 79 |
+
output_dim: 128
|
| 80 |
+
dropout: 0.0
|
| 81 |
+
label:
|
| 82 |
+
enabled: false
|
| 83 |
+
num_classes: 10
|
| 84 |
+
output_dim: 64
|
| 85 |
+
fusion: concat
|
| 86 |
+
condition_dropout: 0.0
|
| 87 |
+
eval:
|
| 88 |
+
n_samples: 10
|
| 89 |
+
metrics:
|
| 90 |
+
- dtw
|
| 91 |
+
- wape
|
| 92 |
+
- ed
|
| 93 |
+
- crps
|
| 94 |
+
- acd
|
| 95 |
+
- sd
|
| 96 |
+
- kd
|
| 97 |
+
- mdd
|
| 98 |
+
- fid
|
| 99 |
+
- sfd
|
| 100 |
+
- jftsd
|
| 101 |
+
- prdc_f1
|
| 102 |
+
- joint_prdc_f1
|
| 103 |
+
- cttp
|
| 104 |
+
- disc_auc
|
| 105 |
+
batch_size: 128
|
| 106 |
+
save_samples: true
|
| 107 |
+
sampler: ddim
|
| 108 |
+
display_interval: 10
|
| 109 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 110 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 111 |
+
cache_folder: __CACHE_ROOT__/tedit/synth-u/seed2
|
| 112 |
+
use_longalign: false
|
| 113 |
+
reference_split: train
|
| 114 |
+
disc_auc:
|
| 115 |
+
k_folds: 5
|
| 116 |
+
epochs: 5
|
| 117 |
+
batch_size: 128
|
| 118 |
+
lr: 0.001
|
| 119 |
+
weight_decay: 0.0001
|
| 120 |
+
hidden_channels: 128
|
| 121 |
+
num_layers: 3
|
| 122 |
+
dropout: 0.1
|
| 123 |
+
seed: 42
|
| 124 |
+
num_workers: 0
|
| 125 |
+
prdc:
|
| 126 |
+
k: 5
|
| 127 |
+
max_samples: 5000
|
| 128 |
+
seed: 0
|
| 129 |
+
metric: euclidean
|
| 130 |
+
backend: auto
|
| 131 |
+
knn_chunk_size: 4096
|
| 132 |
+
joint_enable: true
|
| 133 |
+
joint_weights:
|
| 134 |
+
ts: 1.0
|
| 135 |
+
text: 1.0
|
| 136 |
+
joint_normalize: standard
|
| 137 |
+
segment_classifier:
|
| 138 |
+
enable: false
|
| 139 |
+
segment_len: 128
|
| 140 |
+
n_segments: 3
|
| 141 |
+
seed: 42
|
| 142 |
+
acd_max_lag: 50
|
| 143 |
+
mdd_bins: 32
|
| 144 |
+
use_cache: false
|
| 145 |
+
cache_file: predictions_cache.pkl
|
| 146 |
+
viz:
|
| 147 |
+
enable: true
|
| 148 |
+
k_cases: 20
|
| 149 |
+
max_vars: 8
|
| 150 |
+
ncols: 5
|
| 151 |
+
seed: 42
|
| 152 |
+
dpi: 200
|
| 153 |
+
alpha: 0.25
|
| 154 |
+
figsize_per_subplot:
|
| 155 |
+
- 3.0
|
| 156 |
+
- 2.0
|
| 157 |
+
output_type: pdf
|
| 158 |
+
output_dir: ''
|
| 159 |
+
created_at: '2025-12-29T12:39:13.258262'
|
experiments/tedit/synth-u/seed2/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 5.892986179293807,
|
| 3 |
+
"wape": 97.12165176525684,
|
| 4 |
+
"ed": 9.810233108520508,
|
| 5 |
+
"crps": 0.48175781361963693,
|
| 6 |
+
"acd": 0.06590211391448975,
|
| 7 |
+
"sd": 0.1077936509493189,
|
| 8 |
+
"kd": 0.2053429891488836,
|
| 9 |
+
"mdd": 0.01589180501302083,
|
| 10 |
+
"fid": 41.02142245362586,
|
| 11 |
+
"sfd": 1.7363249259577884,
|
| 12 |
+
"jftsd": 52.48946236949155,
|
| 13 |
+
"prdc_f1.f1": 0.24714336535040912,
|
| 14 |
+
"prdc_f1.precision": 0.54275,
|
| 15 |
+
"prdc_f1.recall": 0.16,
|
| 16 |
+
"joint_prdc_f1.f1": 0.5708573059360731,
|
| 17 |
+
"joint_prdc_f1.precision": 0.73975,
|
| 18 |
+
"joint_prdc_f1.recall": 0.46475,
|
| 19 |
+
"cttp": 21.15283854675293,
|
| 20 |
+
"disc_auc": 0.9766831249999999
|
| 21 |
+
}
|
experiments/tedit/synth-u/seed2/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T12:46:14.889866",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timevqvae/synth-m/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,182 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_timevqvae
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: pretrain
|
| 8 |
+
epochs: 200
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: false
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
- name: finetune
|
| 14 |
+
epochs: 500
|
| 15 |
+
lr: 0.001
|
| 16 |
+
use_condition: true
|
| 17 |
+
freeze_modules: []
|
| 18 |
+
load_from_stage: pretrain
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
stages_preset: two_stage
|
| 21 |
+
epochs: 700
|
| 22 |
+
batch_size: 256
|
| 23 |
+
lr: 0.001
|
| 24 |
+
weight_decay: 0.0001
|
| 25 |
+
scheduler: cosine
|
| 26 |
+
scheduler_params: {}
|
| 27 |
+
early_stopping_patience: 50
|
| 28 |
+
gradient_clip_val: 1.0
|
| 29 |
+
accumulate_grad_batches: 1
|
| 30 |
+
val_check_interval: 1.0
|
| 31 |
+
num_workers: 4
|
| 32 |
+
pin_memory: true
|
| 33 |
+
limit_train_batches: 1.0
|
| 34 |
+
limit_val_batches: 1.0
|
| 35 |
+
limit_test_batches: 1.0
|
| 36 |
+
num_sanity_val_steps: 2
|
| 37 |
+
log_grad_norm: true
|
| 38 |
+
log_param_norm: true
|
| 39 |
+
log_norm_every_n_steps: 50
|
| 40 |
+
data:
|
| 41 |
+
name: synth-m
|
| 42 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 43 |
+
n_var: 2
|
| 44 |
+
seq_length: 128
|
| 45 |
+
normalize: false
|
| 46 |
+
train_split: 0.8
|
| 47 |
+
val_split: 0.1
|
| 48 |
+
model:
|
| 49 |
+
name: timevqvae
|
| 50 |
+
channels: 64
|
| 51 |
+
layers: 3
|
| 52 |
+
nheads: 8
|
| 53 |
+
dropout: 0.1
|
| 54 |
+
vqvae:
|
| 55 |
+
n_fft: 4
|
| 56 |
+
codebook_sizes:
|
| 57 |
+
lf: 1024
|
| 58 |
+
hf: 1024
|
| 59 |
+
codebook_dim: 8
|
| 60 |
+
encoder:
|
| 61 |
+
init_dim: 4
|
| 62 |
+
hid_dim: 128
|
| 63 |
+
n_resnet_blocks: 2
|
| 64 |
+
downsampled_width:
|
| 65 |
+
lf: 8
|
| 66 |
+
hf: 32
|
| 67 |
+
decoder:
|
| 68 |
+
n_resnet_blocks: 2
|
| 69 |
+
maskgit:
|
| 70 |
+
choice_temperatures:
|
| 71 |
+
lf: 10.0
|
| 72 |
+
hf: 0.0
|
| 73 |
+
T:
|
| 74 |
+
lf: 10
|
| 75 |
+
hf: 10
|
| 76 |
+
cfg_scale: 1.0
|
| 77 |
+
ESS:
|
| 78 |
+
use: false
|
| 79 |
+
error_ratio_ma_rate: 0.3
|
| 80 |
+
prior:
|
| 81 |
+
hidden_dim: 128
|
| 82 |
+
n_layers: 4
|
| 83 |
+
heads: 2
|
| 84 |
+
ff_mult: 1.0
|
| 85 |
+
use_rmsnorm: true
|
| 86 |
+
p_unconditional: 0.2
|
| 87 |
+
condition:
|
| 88 |
+
text:
|
| 89 |
+
enabled: false
|
| 90 |
+
input_dim: 1024
|
| 91 |
+
embedding_key: cap_emb
|
| 92 |
+
dropout: 0.0
|
| 93 |
+
cfg_scale: 1.0
|
| 94 |
+
attribute:
|
| 95 |
+
enabled: true
|
| 96 |
+
continuous_dim: 0
|
| 97 |
+
discrete_configs:
|
| 98 |
+
- num_classes: 4
|
| 99 |
+
- num_classes: 2
|
| 100 |
+
- num_classes: 4
|
| 101 |
+
- num_classes: 4
|
| 102 |
+
output_dim: 128
|
| 103 |
+
dropout: 0.0
|
| 104 |
+
label:
|
| 105 |
+
enabled: true
|
| 106 |
+
num_classes: 128
|
| 107 |
+
output_dim: 64
|
| 108 |
+
fusion: concat
|
| 109 |
+
condition_dropout: 0.0
|
| 110 |
+
eval:
|
| 111 |
+
n_samples: 10
|
| 112 |
+
metrics:
|
| 113 |
+
- dtw
|
| 114 |
+
- wape
|
| 115 |
+
- ed
|
| 116 |
+
- crps
|
| 117 |
+
- acd
|
| 118 |
+
- sd
|
| 119 |
+
- kd
|
| 120 |
+
- mdd
|
| 121 |
+
- fid
|
| 122 |
+
- sfd
|
| 123 |
+
- jftsd
|
| 124 |
+
- prdc_f1
|
| 125 |
+
- joint_prdc_f1
|
| 126 |
+
- cttp
|
| 127 |
+
- disc_auc
|
| 128 |
+
batch_size: 128
|
| 129 |
+
save_samples: true
|
| 130 |
+
sampler: ddim
|
| 131 |
+
display_interval: 10
|
| 132 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 133 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 134 |
+
cache_folder: __CACHE_ROOT__/timevqvae/synth-m/seed0
|
| 135 |
+
use_longalign: false
|
| 136 |
+
reference_split: train
|
| 137 |
+
disc_auc:
|
| 138 |
+
k_folds: 5
|
| 139 |
+
epochs: 5
|
| 140 |
+
batch_size: 128
|
| 141 |
+
lr: 0.001
|
| 142 |
+
weight_decay: 0.0001
|
| 143 |
+
hidden_channels: 128
|
| 144 |
+
num_layers: 3
|
| 145 |
+
dropout: 0.1
|
| 146 |
+
seed: 42
|
| 147 |
+
num_workers: 0
|
| 148 |
+
prdc:
|
| 149 |
+
k: 5
|
| 150 |
+
max_samples: 5000
|
| 151 |
+
seed: 0
|
| 152 |
+
metric: euclidean
|
| 153 |
+
backend: auto
|
| 154 |
+
knn_chunk_size: 4096
|
| 155 |
+
joint_enable: true
|
| 156 |
+
joint_weights:
|
| 157 |
+
ts: 1.0
|
| 158 |
+
text: 1.0
|
| 159 |
+
joint_normalize: standard
|
| 160 |
+
segment_classifier:
|
| 161 |
+
enable: false
|
| 162 |
+
segment_len: 128
|
| 163 |
+
n_segments: 3
|
| 164 |
+
seed: 42
|
| 165 |
+
acd_max_lag: 50
|
| 166 |
+
mdd_bins: 32
|
| 167 |
+
use_cache: false
|
| 168 |
+
cache_file: predictions_cache.pkl
|
| 169 |
+
viz:
|
| 170 |
+
enable: true
|
| 171 |
+
k_cases: 20
|
| 172 |
+
max_vars: 8
|
| 173 |
+
ncols: 5
|
| 174 |
+
seed: 42
|
| 175 |
+
dpi: 200
|
| 176 |
+
alpha: 0.25
|
| 177 |
+
figsize_per_subplot:
|
| 178 |
+
- 3.0
|
| 179 |
+
- 2.0
|
| 180 |
+
output_type: pdf
|
| 181 |
+
output_dir: ''
|
| 182 |
+
created_at: '2025-12-29T12:13:04.898091'
|
experiments/timevqvae/synth-m/seed0/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.31858648716823,
|
| 3 |
+
"wape": 81.88684183666123,
|
| 4 |
+
"ed": 12.126584754943847,
|
| 5 |
+
"crps": 0.4874841920208506,
|
| 6 |
+
"acd": 0.07085198909044266,
|
| 7 |
+
"sd": 0.04403555863491975,
|
| 8 |
+
"kd": 0.5307467714991572,
|
| 9 |
+
"mdd": 0.021716542561848955,
|
| 10 |
+
"fid": 60.97225531828698,
|
| 11 |
+
"sfd": 42.101143778819434,
|
| 12 |
+
"jftsd": 67.23506846932088,
|
| 13 |
+
"prdc_f1.f1": 0.09898362892223739,
|
| 14 |
+
"prdc_f1.precision": 0.86375,
|
| 15 |
+
"prdc_f1.recall": 0.0525,
|
| 16 |
+
"joint_prdc_f1.f1": 0.30589631843926995,
|
| 17 |
+
"joint_prdc_f1.precision": 0.58775,
|
| 18 |
+
"joint_prdc_f1.recall": 0.20675,
|
| 19 |
+
"cttp": 19.705974319458008,
|
| 20 |
+
"disc_auc": 0.99985765625
|
| 21 |
+
}
|
experiments/timevqvae/synth-m/seed0/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T13:38:29.052592",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timevqvae/synth-m/seed1/config.template.yaml
ADDED
|
@@ -0,0 +1,182 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_timevqvae
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 1
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: pretrain
|
| 8 |
+
epochs: 200
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: false
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
- name: finetune
|
| 14 |
+
epochs: 500
|
| 15 |
+
lr: 0.001
|
| 16 |
+
use_condition: true
|
| 17 |
+
freeze_modules: []
|
| 18 |
+
load_from_stage: pretrain
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
stages_preset: two_stage
|
| 21 |
+
epochs: 700
|
| 22 |
+
batch_size: 256
|
| 23 |
+
lr: 0.001
|
| 24 |
+
weight_decay: 0.0001
|
| 25 |
+
scheduler: cosine
|
| 26 |
+
scheduler_params: {}
|
| 27 |
+
early_stopping_patience: 50
|
| 28 |
+
gradient_clip_val: 1.0
|
| 29 |
+
accumulate_grad_batches: 1
|
| 30 |
+
val_check_interval: 1.0
|
| 31 |
+
num_workers: 4
|
| 32 |
+
pin_memory: true
|
| 33 |
+
limit_train_batches: 1.0
|
| 34 |
+
limit_val_batches: 1.0
|
| 35 |
+
limit_test_batches: 1.0
|
| 36 |
+
num_sanity_val_steps: 2
|
| 37 |
+
log_grad_norm: true
|
| 38 |
+
log_param_norm: true
|
| 39 |
+
log_norm_every_n_steps: 50
|
| 40 |
+
data:
|
| 41 |
+
name: synth-m
|
| 42 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 43 |
+
n_var: 2
|
| 44 |
+
seq_length: 128
|
| 45 |
+
normalize: false
|
| 46 |
+
train_split: 0.8
|
| 47 |
+
val_split: 0.1
|
| 48 |
+
model:
|
| 49 |
+
name: timevqvae
|
| 50 |
+
channels: 64
|
| 51 |
+
layers: 3
|
| 52 |
+
nheads: 8
|
| 53 |
+
dropout: 0.1
|
| 54 |
+
vqvae:
|
| 55 |
+
n_fft: 4
|
| 56 |
+
codebook_sizes:
|
| 57 |
+
lf: 1024
|
| 58 |
+
hf: 1024
|
| 59 |
+
codebook_dim: 8
|
| 60 |
+
encoder:
|
| 61 |
+
init_dim: 4
|
| 62 |
+
hid_dim: 128
|
| 63 |
+
n_resnet_blocks: 2
|
| 64 |
+
downsampled_width:
|
| 65 |
+
lf: 8
|
| 66 |
+
hf: 32
|
| 67 |
+
decoder:
|
| 68 |
+
n_resnet_blocks: 2
|
| 69 |
+
maskgit:
|
| 70 |
+
choice_temperatures:
|
| 71 |
+
lf: 10.0
|
| 72 |
+
hf: 0.0
|
| 73 |
+
T:
|
| 74 |
+
lf: 10
|
| 75 |
+
hf: 10
|
| 76 |
+
cfg_scale: 1.0
|
| 77 |
+
ESS:
|
| 78 |
+
use: false
|
| 79 |
+
error_ratio_ma_rate: 0.3
|
| 80 |
+
prior:
|
| 81 |
+
hidden_dim: 128
|
| 82 |
+
n_layers: 4
|
| 83 |
+
heads: 2
|
| 84 |
+
ff_mult: 1.0
|
| 85 |
+
use_rmsnorm: true
|
| 86 |
+
p_unconditional: 0.2
|
| 87 |
+
condition:
|
| 88 |
+
text:
|
| 89 |
+
enabled: false
|
| 90 |
+
input_dim: 1024
|
| 91 |
+
embedding_key: cap_emb
|
| 92 |
+
dropout: 0.0
|
| 93 |
+
cfg_scale: 1.0
|
| 94 |
+
attribute:
|
| 95 |
+
enabled: true
|
| 96 |
+
continuous_dim: 0
|
| 97 |
+
discrete_configs:
|
| 98 |
+
- num_classes: 4
|
| 99 |
+
- num_classes: 2
|
| 100 |
+
- num_classes: 4
|
| 101 |
+
- num_classes: 4
|
| 102 |
+
output_dim: 128
|
| 103 |
+
dropout: 0.0
|
| 104 |
+
label:
|
| 105 |
+
enabled: true
|
| 106 |
+
num_classes: 128
|
| 107 |
+
output_dim: 64
|
| 108 |
+
fusion: concat
|
| 109 |
+
condition_dropout: 0.0
|
| 110 |
+
eval:
|
| 111 |
+
n_samples: 10
|
| 112 |
+
metrics:
|
| 113 |
+
- dtw
|
| 114 |
+
- wape
|
| 115 |
+
- ed
|
| 116 |
+
- crps
|
| 117 |
+
- acd
|
| 118 |
+
- sd
|
| 119 |
+
- kd
|
| 120 |
+
- mdd
|
| 121 |
+
- fid
|
| 122 |
+
- sfd
|
| 123 |
+
- jftsd
|
| 124 |
+
- prdc_f1
|
| 125 |
+
- joint_prdc_f1
|
| 126 |
+
- cttp
|
| 127 |
+
- disc_auc
|
| 128 |
+
batch_size: 128
|
| 129 |
+
save_samples: true
|
| 130 |
+
sampler: ddim
|
| 131 |
+
display_interval: 10
|
| 132 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 133 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 134 |
+
cache_folder: __CACHE_ROOT__/timevqvae/synth-m/seed1
|
| 135 |
+
use_longalign: false
|
| 136 |
+
reference_split: train
|
| 137 |
+
disc_auc:
|
| 138 |
+
k_folds: 5
|
| 139 |
+
epochs: 5
|
| 140 |
+
batch_size: 128
|
| 141 |
+
lr: 0.001
|
| 142 |
+
weight_decay: 0.0001
|
| 143 |
+
hidden_channels: 128
|
| 144 |
+
num_layers: 3
|
| 145 |
+
dropout: 0.1
|
| 146 |
+
seed: 42
|
| 147 |
+
num_workers: 0
|
| 148 |
+
prdc:
|
| 149 |
+
k: 5
|
| 150 |
+
max_samples: 5000
|
| 151 |
+
seed: 0
|
| 152 |
+
metric: euclidean
|
| 153 |
+
backend: auto
|
| 154 |
+
knn_chunk_size: 4096
|
| 155 |
+
joint_enable: true
|
| 156 |
+
joint_weights:
|
| 157 |
+
ts: 1.0
|
| 158 |
+
text: 1.0
|
| 159 |
+
joint_normalize: standard
|
| 160 |
+
segment_classifier:
|
| 161 |
+
enable: false
|
| 162 |
+
segment_len: 128
|
| 163 |
+
n_segments: 3
|
| 164 |
+
seed: 42
|
| 165 |
+
acd_max_lag: 50
|
| 166 |
+
mdd_bins: 32
|
| 167 |
+
use_cache: false
|
| 168 |
+
cache_file: predictions_cache.pkl
|
| 169 |
+
viz:
|
| 170 |
+
enable: true
|
| 171 |
+
k_cases: 20
|
| 172 |
+
max_vars: 8
|
| 173 |
+
ncols: 5
|
| 174 |
+
seed: 42
|
| 175 |
+
dpi: 200
|
| 176 |
+
alpha: 0.25
|
| 177 |
+
figsize_per_subplot:
|
| 178 |
+
- 3.0
|
| 179 |
+
- 2.0
|
| 180 |
+
output_type: pdf
|
| 181 |
+
output_dir: ''
|
| 182 |
+
created_at: '2025-12-29T12:13:36.367851'
|
experiments/timevqvae/synth-m/seed1/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.343814207994294,
|
| 3 |
+
"wape": 82.52775025941148,
|
| 4 |
+
"ed": 12.224420776367188,
|
| 5 |
+
"crps": 0.4997049319745764,
|
| 6 |
+
"acd": 0.07632005959749222,
|
| 7 |
+
"sd": 0.025823204978308596,
|
| 8 |
+
"kd": 0.6081971666035935,
|
| 9 |
+
"mdd": 0.020432098388671868,
|
| 10 |
+
"fid": 60.14093743514633,
|
| 11 |
+
"sfd": 53.61667895938183,
|
| 12 |
+
"jftsd": 66.51047358008486,
|
| 13 |
+
"prdc_f1.f1": 0.10702228187919463,
|
| 14 |
+
"prdc_f1.precision": 0.87425,
|
| 15 |
+
"prdc_f1.recall": 0.057,
|
| 16 |
+
"joint_prdc_f1.f1": 0.31592179289026273,
|
| 17 |
+
"joint_prdc_f1.precision": 0.5935,
|
| 18 |
+
"joint_prdc_f1.recall": 0.21525,
|
| 19 |
+
"cttp": 19.920206237792968,
|
| 20 |
+
"disc_auc": 0.9996260937500001
|
| 21 |
+
}
|
experiments/timevqvae/synth-m/seed1/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T13:53:15.114698",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timevqvae/synth-m/seed2/config.template.yaml
ADDED
|
@@ -0,0 +1,182 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_timevqvae
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 2
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: pretrain
|
| 8 |
+
epochs: 200
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: false
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
- name: finetune
|
| 14 |
+
epochs: 500
|
| 15 |
+
lr: 0.001
|
| 16 |
+
use_condition: true
|
| 17 |
+
freeze_modules: []
|
| 18 |
+
load_from_stage: pretrain
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
stages_preset: two_stage
|
| 21 |
+
epochs: 700
|
| 22 |
+
batch_size: 256
|
| 23 |
+
lr: 0.001
|
| 24 |
+
weight_decay: 0.0001
|
| 25 |
+
scheduler: cosine
|
| 26 |
+
scheduler_params: {}
|
| 27 |
+
early_stopping_patience: 50
|
| 28 |
+
gradient_clip_val: 1.0
|
| 29 |
+
accumulate_grad_batches: 1
|
| 30 |
+
val_check_interval: 1.0
|
| 31 |
+
num_workers: 4
|
| 32 |
+
pin_memory: true
|
| 33 |
+
limit_train_batches: 1.0
|
| 34 |
+
limit_val_batches: 1.0
|
| 35 |
+
limit_test_batches: 1.0
|
| 36 |
+
num_sanity_val_steps: 2
|
| 37 |
+
log_grad_norm: true
|
| 38 |
+
log_param_norm: true
|
| 39 |
+
log_norm_every_n_steps: 50
|
| 40 |
+
data:
|
| 41 |
+
name: synth-m
|
| 42 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 43 |
+
n_var: 2
|
| 44 |
+
seq_length: 128
|
| 45 |
+
normalize: false
|
| 46 |
+
train_split: 0.8
|
| 47 |
+
val_split: 0.1
|
| 48 |
+
model:
|
| 49 |
+
name: timevqvae
|
| 50 |
+
channels: 64
|
| 51 |
+
layers: 3
|
| 52 |
+
nheads: 8
|
| 53 |
+
dropout: 0.1
|
| 54 |
+
vqvae:
|
| 55 |
+
n_fft: 4
|
| 56 |
+
codebook_sizes:
|
| 57 |
+
lf: 1024
|
| 58 |
+
hf: 1024
|
| 59 |
+
codebook_dim: 8
|
| 60 |
+
encoder:
|
| 61 |
+
init_dim: 4
|
| 62 |
+
hid_dim: 128
|
| 63 |
+
n_resnet_blocks: 2
|
| 64 |
+
downsampled_width:
|
| 65 |
+
lf: 8
|
| 66 |
+
hf: 32
|
| 67 |
+
decoder:
|
| 68 |
+
n_resnet_blocks: 2
|
| 69 |
+
maskgit:
|
| 70 |
+
choice_temperatures:
|
| 71 |
+
lf: 10.0
|
| 72 |
+
hf: 0.0
|
| 73 |
+
T:
|
| 74 |
+
lf: 10
|
| 75 |
+
hf: 10
|
| 76 |
+
cfg_scale: 1.0
|
| 77 |
+
ESS:
|
| 78 |
+
use: false
|
| 79 |
+
error_ratio_ma_rate: 0.3
|
| 80 |
+
prior:
|
| 81 |
+
hidden_dim: 128
|
| 82 |
+
n_layers: 4
|
| 83 |
+
heads: 2
|
| 84 |
+
ff_mult: 1.0
|
| 85 |
+
use_rmsnorm: true
|
| 86 |
+
p_unconditional: 0.2
|
| 87 |
+
condition:
|
| 88 |
+
text:
|
| 89 |
+
enabled: false
|
| 90 |
+
input_dim: 1024
|
| 91 |
+
embedding_key: cap_emb
|
| 92 |
+
dropout: 0.0
|
| 93 |
+
cfg_scale: 1.0
|
| 94 |
+
attribute:
|
| 95 |
+
enabled: true
|
| 96 |
+
continuous_dim: 0
|
| 97 |
+
discrete_configs:
|
| 98 |
+
- num_classes: 4
|
| 99 |
+
- num_classes: 2
|
| 100 |
+
- num_classes: 4
|
| 101 |
+
- num_classes: 4
|
| 102 |
+
output_dim: 128
|
| 103 |
+
dropout: 0.0
|
| 104 |
+
label:
|
| 105 |
+
enabled: true
|
| 106 |
+
num_classes: 128
|
| 107 |
+
output_dim: 64
|
| 108 |
+
fusion: concat
|
| 109 |
+
condition_dropout: 0.0
|
| 110 |
+
eval:
|
| 111 |
+
n_samples: 10
|
| 112 |
+
metrics:
|
| 113 |
+
- dtw
|
| 114 |
+
- wape
|
| 115 |
+
- ed
|
| 116 |
+
- crps
|
| 117 |
+
- acd
|
| 118 |
+
- sd
|
| 119 |
+
- kd
|
| 120 |
+
- mdd
|
| 121 |
+
- fid
|
| 122 |
+
- sfd
|
| 123 |
+
- jftsd
|
| 124 |
+
- prdc_f1
|
| 125 |
+
- joint_prdc_f1
|
| 126 |
+
- cttp
|
| 127 |
+
- disc_auc
|
| 128 |
+
batch_size: 128
|
| 129 |
+
save_samples: true
|
| 130 |
+
sampler: ddim
|
| 131 |
+
display_interval: 10
|
| 132 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 133 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 134 |
+
cache_folder: __CACHE_ROOT__/timevqvae/synth-m/seed2
|
| 135 |
+
use_longalign: false
|
| 136 |
+
reference_split: train
|
| 137 |
+
disc_auc:
|
| 138 |
+
k_folds: 5
|
| 139 |
+
epochs: 5
|
| 140 |
+
batch_size: 128
|
| 141 |
+
lr: 0.001
|
| 142 |
+
weight_decay: 0.0001
|
| 143 |
+
hidden_channels: 128
|
| 144 |
+
num_layers: 3
|
| 145 |
+
dropout: 0.1
|
| 146 |
+
seed: 42
|
| 147 |
+
num_workers: 0
|
| 148 |
+
prdc:
|
| 149 |
+
k: 5
|
| 150 |
+
max_samples: 5000
|
| 151 |
+
seed: 0
|
| 152 |
+
metric: euclidean
|
| 153 |
+
backend: auto
|
| 154 |
+
knn_chunk_size: 4096
|
| 155 |
+
joint_enable: true
|
| 156 |
+
joint_weights:
|
| 157 |
+
ts: 1.0
|
| 158 |
+
text: 1.0
|
| 159 |
+
joint_normalize: standard
|
| 160 |
+
segment_classifier:
|
| 161 |
+
enable: false
|
| 162 |
+
segment_len: 128
|
| 163 |
+
n_segments: 3
|
| 164 |
+
seed: 42
|
| 165 |
+
acd_max_lag: 50
|
| 166 |
+
mdd_bins: 32
|
| 167 |
+
use_cache: false
|
| 168 |
+
cache_file: predictions_cache.pkl
|
| 169 |
+
viz:
|
| 170 |
+
enable: true
|
| 171 |
+
k_cases: 20
|
| 172 |
+
max_vars: 8
|
| 173 |
+
ncols: 5
|
| 174 |
+
seed: 42
|
| 175 |
+
dpi: 200
|
| 176 |
+
alpha: 0.25
|
| 177 |
+
figsize_per_subplot:
|
| 178 |
+
- 3.0
|
| 179 |
+
- 2.0
|
| 180 |
+
output_type: pdf
|
| 181 |
+
output_dir: ''
|
| 182 |
+
created_at: '2025-12-29T12:17:07.015654'
|
experiments/timevqvae/synth-m/seed2/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.057030897705099,
|
| 3 |
+
"wape": 80.26683620723146,
|
| 4 |
+
"ed": 11.922863044738769,
|
| 5 |
+
"crps": 0.47646317756947987,
|
| 6 |
+
"acd": 0.07277964800596237,
|
| 7 |
+
"sd": 0.014800543923988174,
|
| 8 |
+
"kd": 0.7531600463214593,
|
| 9 |
+
"mdd": 0.021204711914062493,
|
| 10 |
+
"fid": 60.74959109588518,
|
| 11 |
+
"sfd": 47.06639864320983,
|
| 12 |
+
"jftsd": 66.83946958109215,
|
| 13 |
+
"prdc_f1.f1": 0.09407817634173056,
|
| 14 |
+
"prdc_f1.precision": 0.86325,
|
| 15 |
+
"prdc_f1.recall": 0.04975,
|
| 16 |
+
"joint_prdc_f1.f1": 0.31364056720098643,
|
| 17 |
+
"joint_prdc_f1.precision": 0.5985,
|
| 18 |
+
"joint_prdc_f1.recall": 0.2125,
|
| 19 |
+
"cttp": 19.964169830322266,
|
| 20 |
+
"disc_auc": 0.999586875
|
| 21 |
+
}
|
experiments/timevqvae/synth-m/seed2/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-29T13:48:00.124971",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timevqvae/synth-u/seed2/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 4.787562399860754,
|
| 3 |
+
"wape": 78.679489850362,
|
| 4 |
+
"ed": 8.096774627685546,
|
| 5 |
+
"crps": 0.46088996795434334,
|
| 6 |
+
"acd": 0.08149992674589157,
|
| 7 |
+
"sd": 0.06299340445322331,
|
| 8 |
+
"kd": 0.759065214899195,
|
| 9 |
+
"mdd": 0.025206624348958328,
|
| 10 |
+
"fid": 77.73334841368484,
|
| 11 |
+
"sfd": 5.412455164232835,
|
| 12 |
+
"jftsd": 85.91943105891795,
|
| 13 |
+
"prdc_f1.f1": 0.035607669122572,
|
| 14 |
+
"prdc_f1.precision": 0.72825,
|
| 15 |
+
"prdc_f1.recall": 0.01825,
|
| 16 |
+
"joint_prdc_f1.f1": 0.31903679653679656,
|
| 17 |
+
"joint_prdc_f1.precision": 0.719,
|
| 18 |
+
"joint_prdc_f1.recall": 0.205,
|
| 19 |
+
"cttp": 15.809910827636719,
|
| 20 |
+
"disc_auc": 0.9997239062500001
|
| 21 |
+
}
|
experiments/timeweaver/synth-m/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,155 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_timeweaver
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: timeweaver
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
attr_dim: 64
|
| 47 |
+
num_attr_heads: 4
|
| 48 |
+
diffusion_steps: 50
|
| 49 |
+
noise_schedule: quad
|
| 50 |
+
beta_start: 0.0001
|
| 51 |
+
beta_end: 0.5
|
| 52 |
+
base_patch: 4
|
| 53 |
+
attr_proj_type: avg
|
| 54 |
+
attr_heads: 8
|
| 55 |
+
attr_layers: 2
|
| 56 |
+
n_attrs: 4
|
| 57 |
+
feature_ratio: 0.5
|
| 58 |
+
side_var_emb: 16
|
| 59 |
+
side_time_emb: 112
|
| 60 |
+
condition:
|
| 61 |
+
text:
|
| 62 |
+
enabled: false
|
| 63 |
+
input_dim: 1024
|
| 64 |
+
embedding_key: cap_emb
|
| 65 |
+
dropout: 0.0
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: true
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs:
|
| 71 |
+
- num_classes: 4
|
| 72 |
+
- num_classes: 2
|
| 73 |
+
- num_classes: 4
|
| 74 |
+
- num_classes: 4
|
| 75 |
+
output_dim: 128
|
| 76 |
+
dropout: 0.0
|
| 77 |
+
label:
|
| 78 |
+
enabled: false
|
| 79 |
+
num_classes: 10
|
| 80 |
+
output_dim: 64
|
| 81 |
+
fusion: concat
|
| 82 |
+
condition_dropout: 0.0
|
| 83 |
+
eval:
|
| 84 |
+
n_samples: 10
|
| 85 |
+
metrics:
|
| 86 |
+
- dtw
|
| 87 |
+
- wape
|
| 88 |
+
- ed
|
| 89 |
+
- crps
|
| 90 |
+
- acd
|
| 91 |
+
- sd
|
| 92 |
+
- kd
|
| 93 |
+
- mdd
|
| 94 |
+
- fid
|
| 95 |
+
- sfd
|
| 96 |
+
- jftsd
|
| 97 |
+
- prdc_f1
|
| 98 |
+
- joint_prdc_f1
|
| 99 |
+
- cttp
|
| 100 |
+
- disc_auc
|
| 101 |
+
batch_size: 128
|
| 102 |
+
save_samples: true
|
| 103 |
+
sampler: ddim
|
| 104 |
+
display_interval: 10
|
| 105 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 106 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 107 |
+
cache_folder: __CACHE_ROOT__/timeweaver/synth-m/seed0
|
| 108 |
+
use_longalign: false
|
| 109 |
+
reference_split: train
|
| 110 |
+
disc_auc:
|
| 111 |
+
k_folds: 5
|
| 112 |
+
epochs: 5
|
| 113 |
+
batch_size: 128
|
| 114 |
+
lr: 0.001
|
| 115 |
+
weight_decay: 0.0001
|
| 116 |
+
hidden_channels: 128
|
| 117 |
+
num_layers: 3
|
| 118 |
+
dropout: 0.1
|
| 119 |
+
seed: 42
|
| 120 |
+
num_workers: 0
|
| 121 |
+
prdc:
|
| 122 |
+
k: 5
|
| 123 |
+
max_samples: 5000
|
| 124 |
+
seed: 0
|
| 125 |
+
metric: euclidean
|
| 126 |
+
backend: auto
|
| 127 |
+
knn_chunk_size: 4096
|
| 128 |
+
joint_enable: true
|
| 129 |
+
joint_weights:
|
| 130 |
+
ts: 1.0
|
| 131 |
+
text: 1.0
|
| 132 |
+
joint_normalize: standard
|
| 133 |
+
segment_classifier:
|
| 134 |
+
enable: false
|
| 135 |
+
segment_len: 128
|
| 136 |
+
n_segments: 3
|
| 137 |
+
seed: 42
|
| 138 |
+
acd_max_lag: 50
|
| 139 |
+
mdd_bins: 32
|
| 140 |
+
use_cache: false
|
| 141 |
+
cache_file: predictions_cache.pkl
|
| 142 |
+
viz:
|
| 143 |
+
enable: true
|
| 144 |
+
k_cases: 20
|
| 145 |
+
max_vars: 8
|
| 146 |
+
ncols: 5
|
| 147 |
+
seed: 42
|
| 148 |
+
dpi: 200
|
| 149 |
+
alpha: 0.25
|
| 150 |
+
figsize_per_subplot:
|
| 151 |
+
- 3.0
|
| 152 |
+
- 2.0
|
| 153 |
+
output_type: pdf
|
| 154 |
+
output_dir: ''
|
| 155 |
+
created_at: '2025-12-27T20:26:50.198854'
|
experiments/timeweaver/synth-m/seed0/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T21:14:55.616926",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timeweaver/synth-u/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,154 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_timeweaver
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: timeweaver
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
attr_dim: 64
|
| 47 |
+
num_attr_heads: 4
|
| 48 |
+
diffusion_steps: 50
|
| 49 |
+
noise_schedule: quad
|
| 50 |
+
beta_start: 0.0001
|
| 51 |
+
beta_end: 0.5
|
| 52 |
+
base_patch: 4
|
| 53 |
+
attr_proj_type: avg
|
| 54 |
+
attr_heads: 8
|
| 55 |
+
attr_layers: 2
|
| 56 |
+
n_attrs: 4
|
| 57 |
+
feature_ratio: 0.5
|
| 58 |
+
side_var_emb: 16
|
| 59 |
+
side_time_emb: 112
|
| 60 |
+
condition:
|
| 61 |
+
text:
|
| 62 |
+
enabled: false
|
| 63 |
+
input_dim: 1024
|
| 64 |
+
embedding_key: cap_emb
|
| 65 |
+
dropout: 0.0
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: true
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs:
|
| 71 |
+
- num_classes: 4
|
| 72 |
+
- num_classes: 2
|
| 73 |
+
- num_classes: 4
|
| 74 |
+
output_dim: 128
|
| 75 |
+
dropout: 0.0
|
| 76 |
+
label:
|
| 77 |
+
enabled: false
|
| 78 |
+
num_classes: 10
|
| 79 |
+
output_dim: 64
|
| 80 |
+
fusion: concat
|
| 81 |
+
condition_dropout: 0.0
|
| 82 |
+
eval:
|
| 83 |
+
n_samples: 10
|
| 84 |
+
metrics:
|
| 85 |
+
- dtw
|
| 86 |
+
- wape
|
| 87 |
+
- ed
|
| 88 |
+
- crps
|
| 89 |
+
- acd
|
| 90 |
+
- sd
|
| 91 |
+
- kd
|
| 92 |
+
- mdd
|
| 93 |
+
- fid
|
| 94 |
+
- sfd
|
| 95 |
+
- jftsd
|
| 96 |
+
- prdc_f1
|
| 97 |
+
- joint_prdc_f1
|
| 98 |
+
- cttp
|
| 99 |
+
- disc_auc
|
| 100 |
+
batch_size: 128
|
| 101 |
+
save_samples: true
|
| 102 |
+
sampler: ddim
|
| 103 |
+
display_interval: 10
|
| 104 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 105 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 106 |
+
cache_folder: __CACHE_ROOT__/timeweaver/synth-u/seed0
|
| 107 |
+
use_longalign: false
|
| 108 |
+
reference_split: train
|
| 109 |
+
disc_auc:
|
| 110 |
+
k_folds: 5
|
| 111 |
+
epochs: 5
|
| 112 |
+
batch_size: 128
|
| 113 |
+
lr: 0.001
|
| 114 |
+
weight_decay: 0.0001
|
| 115 |
+
hidden_channels: 128
|
| 116 |
+
num_layers: 3
|
| 117 |
+
dropout: 0.1
|
| 118 |
+
seed: 42
|
| 119 |
+
num_workers: 0
|
| 120 |
+
prdc:
|
| 121 |
+
k: 5
|
| 122 |
+
max_samples: 5000
|
| 123 |
+
seed: 0
|
| 124 |
+
metric: euclidean
|
| 125 |
+
backend: auto
|
| 126 |
+
knn_chunk_size: 4096
|
| 127 |
+
joint_enable: true
|
| 128 |
+
joint_weights:
|
| 129 |
+
ts: 1.0
|
| 130 |
+
text: 1.0
|
| 131 |
+
joint_normalize: standard
|
| 132 |
+
segment_classifier:
|
| 133 |
+
enable: false
|
| 134 |
+
segment_len: 128
|
| 135 |
+
n_segments: 3
|
| 136 |
+
seed: 42
|
| 137 |
+
acd_max_lag: 50
|
| 138 |
+
mdd_bins: 32
|
| 139 |
+
use_cache: false
|
| 140 |
+
cache_file: predictions_cache.pkl
|
| 141 |
+
viz:
|
| 142 |
+
enable: true
|
| 143 |
+
k_cases: 20
|
| 144 |
+
max_vars: 8
|
| 145 |
+
ncols: 5
|
| 146 |
+
seed: 42
|
| 147 |
+
dpi: 200
|
| 148 |
+
alpha: 0.25
|
| 149 |
+
figsize_per_subplot:
|
| 150 |
+
- 3.0
|
| 151 |
+
- 2.0
|
| 152 |
+
output_type: pdf
|
| 153 |
+
output_dir: ''
|
| 154 |
+
created_at: '2025-12-27T20:37:00.990690'
|
experiments/timeweaver/synth-u/seed0/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 6.5257270444318785,
|
| 3 |
+
"wape": 106.23219971334166,
|
| 4 |
+
"ed": 10.546296043395996,
|
| 5 |
+
"crps": 0.6598063489456146,
|
| 6 |
+
"acd": 0.0835137665271759,
|
| 7 |
+
"sd": 0.03484983097586652,
|
| 8 |
+
"kd": 0.7245602803386209,
|
| 9 |
+
"mdd": 0.02627673339843749,
|
| 10 |
+
"fid": 71.69699309532854,
|
| 11 |
+
"sfd": 2.824899052929652,
|
| 12 |
+
"jftsd": 81.25624647681377,
|
| 13 |
+
"prdc_f1.f1": 0.05651001430615164,
|
| 14 |
+
"prdc_f1.precision": 0.6695,
|
| 15 |
+
"prdc_f1.recall": 0.0295,
|
| 16 |
+
"joint_prdc_f1.f1": 0.3397619872136388,
|
| 17 |
+
"joint_prdc_f1.precision": 0.71575,
|
| 18 |
+
"joint_prdc_f1.recall": 0.22275,
|
| 19 |
+
"cttp": 17.022034454345704,
|
| 20 |
+
"disc_auc": 0.9896035937500001
|
| 21 |
+
}
|
experiments/timeweaver/synth-u/seed0/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T20:50:28.675173",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timeweaver/synth-u/seed1/config.template.yaml
ADDED
|
@@ -0,0 +1,154 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_timeweaver
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 1
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: timeweaver
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
attr_dim: 64
|
| 47 |
+
num_attr_heads: 4
|
| 48 |
+
diffusion_steps: 50
|
| 49 |
+
noise_schedule: quad
|
| 50 |
+
beta_start: 0.0001
|
| 51 |
+
beta_end: 0.5
|
| 52 |
+
base_patch: 4
|
| 53 |
+
attr_proj_type: avg
|
| 54 |
+
attr_heads: 8
|
| 55 |
+
attr_layers: 2
|
| 56 |
+
n_attrs: 4
|
| 57 |
+
feature_ratio: 0.5
|
| 58 |
+
side_var_emb: 16
|
| 59 |
+
side_time_emb: 112
|
| 60 |
+
condition:
|
| 61 |
+
text:
|
| 62 |
+
enabled: false
|
| 63 |
+
input_dim: 1024
|
| 64 |
+
embedding_key: cap_emb
|
| 65 |
+
dropout: 0.0
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: true
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs:
|
| 71 |
+
- num_classes: 4
|
| 72 |
+
- num_classes: 2
|
| 73 |
+
- num_classes: 4
|
| 74 |
+
output_dim: 128
|
| 75 |
+
dropout: 0.0
|
| 76 |
+
label:
|
| 77 |
+
enabled: false
|
| 78 |
+
num_classes: 10
|
| 79 |
+
output_dim: 64
|
| 80 |
+
fusion: concat
|
| 81 |
+
condition_dropout: 0.0
|
| 82 |
+
eval:
|
| 83 |
+
n_samples: 10
|
| 84 |
+
metrics:
|
| 85 |
+
- dtw
|
| 86 |
+
- wape
|
| 87 |
+
- ed
|
| 88 |
+
- crps
|
| 89 |
+
- acd
|
| 90 |
+
- sd
|
| 91 |
+
- kd
|
| 92 |
+
- mdd
|
| 93 |
+
- fid
|
| 94 |
+
- sfd
|
| 95 |
+
- jftsd
|
| 96 |
+
- prdc_f1
|
| 97 |
+
- joint_prdc_f1
|
| 98 |
+
- cttp
|
| 99 |
+
- disc_auc
|
| 100 |
+
batch_size: 128
|
| 101 |
+
save_samples: true
|
| 102 |
+
sampler: ddim
|
| 103 |
+
display_interval: 10
|
| 104 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 105 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 106 |
+
cache_folder: __CACHE_ROOT__/timeweaver/synth-u/seed1
|
| 107 |
+
use_longalign: false
|
| 108 |
+
reference_split: train
|
| 109 |
+
disc_auc:
|
| 110 |
+
k_folds: 5
|
| 111 |
+
epochs: 5
|
| 112 |
+
batch_size: 128
|
| 113 |
+
lr: 0.001
|
| 114 |
+
weight_decay: 0.0001
|
| 115 |
+
hidden_channels: 128
|
| 116 |
+
num_layers: 3
|
| 117 |
+
dropout: 0.1
|
| 118 |
+
seed: 42
|
| 119 |
+
num_workers: 0
|
| 120 |
+
prdc:
|
| 121 |
+
k: 5
|
| 122 |
+
max_samples: 5000
|
| 123 |
+
seed: 0
|
| 124 |
+
metric: euclidean
|
| 125 |
+
backend: auto
|
| 126 |
+
knn_chunk_size: 4096
|
| 127 |
+
joint_enable: true
|
| 128 |
+
joint_weights:
|
| 129 |
+
ts: 1.0
|
| 130 |
+
text: 1.0
|
| 131 |
+
joint_normalize: standard
|
| 132 |
+
segment_classifier:
|
| 133 |
+
enable: false
|
| 134 |
+
segment_len: 128
|
| 135 |
+
n_segments: 3
|
| 136 |
+
seed: 42
|
| 137 |
+
acd_max_lag: 50
|
| 138 |
+
mdd_bins: 32
|
| 139 |
+
use_cache: false
|
| 140 |
+
cache_file: predictions_cache.pkl
|
| 141 |
+
viz:
|
| 142 |
+
enable: true
|
| 143 |
+
k_cases: 20
|
| 144 |
+
max_vars: 8
|
| 145 |
+
ncols: 5
|
| 146 |
+
seed: 42
|
| 147 |
+
dpi: 200
|
| 148 |
+
alpha: 0.25
|
| 149 |
+
figsize_per_subplot:
|
| 150 |
+
- 3.0
|
| 151 |
+
- 2.0
|
| 152 |
+
output_type: pdf
|
| 153 |
+
output_dir: ''
|
| 154 |
+
created_at: '2025-12-27T20:38:04.132978'
|
experiments/timeweaver/synth-u/seed1/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 5.083456673753537,
|
| 3 |
+
"wape": 86.62641059459605,
|
| 4 |
+
"ed": 8.87348338317871,
|
| 5 |
+
"crps": 0.4577799159935779,
|
| 6 |
+
"acd": 0.06394154578447342,
|
| 7 |
+
"sd": 0.03798199907397587,
|
| 8 |
+
"kd": 0.40341618695346027,
|
| 9 |
+
"mdd": 0.014729980468749997,
|
| 10 |
+
"fid": 42.469093827007754,
|
| 11 |
+
"sfd": 1.5728544362453079,
|
| 12 |
+
"jftsd": 53.49644161443217,
|
| 13 |
+
"prdc_f1.f1": 0.2789435205522435,
|
| 14 |
+
"prdc_f1.precision": 0.6165,
|
| 15 |
+
"prdc_f1.recall": 0.18025,
|
| 16 |
+
"joint_prdc_f1.f1": 0.56967758444217,
|
| 17 |
+
"joint_prdc_f1.precision": 0.76875,
|
| 18 |
+
"joint_prdc_f1.recall": 0.4525,
|
| 19 |
+
"cttp": 20.765921737670897,
|
| 20 |
+
"disc_auc": 0.97113171875
|
| 21 |
+
}
|
experiments/timeweaver/synth-u/seed1/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T20:51:39.026544",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/timeweaver/synth-u/seed2/config.template.yaml
ADDED
|
@@ -0,0 +1,154 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_timeweaver
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 2
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: timeweaver
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
attr_dim: 64
|
| 47 |
+
num_attr_heads: 4
|
| 48 |
+
diffusion_steps: 50
|
| 49 |
+
noise_schedule: quad
|
| 50 |
+
beta_start: 0.0001
|
| 51 |
+
beta_end: 0.5
|
| 52 |
+
base_patch: 4
|
| 53 |
+
attr_proj_type: avg
|
| 54 |
+
attr_heads: 8
|
| 55 |
+
attr_layers: 2
|
| 56 |
+
n_attrs: 4
|
| 57 |
+
feature_ratio: 0.5
|
| 58 |
+
side_var_emb: 16
|
| 59 |
+
side_time_emb: 112
|
| 60 |
+
condition:
|
| 61 |
+
text:
|
| 62 |
+
enabled: false
|
| 63 |
+
input_dim: 1024
|
| 64 |
+
embedding_key: cap_emb
|
| 65 |
+
dropout: 0.0
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: true
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs:
|
| 71 |
+
- num_classes: 4
|
| 72 |
+
- num_classes: 2
|
| 73 |
+
- num_classes: 4
|
| 74 |
+
output_dim: 128
|
| 75 |
+
dropout: 0.0
|
| 76 |
+
label:
|
| 77 |
+
enabled: false
|
| 78 |
+
num_classes: 10
|
| 79 |
+
output_dim: 64
|
| 80 |
+
fusion: concat
|
| 81 |
+
condition_dropout: 0.0
|
| 82 |
+
eval:
|
| 83 |
+
n_samples: 10
|
| 84 |
+
metrics:
|
| 85 |
+
- dtw
|
| 86 |
+
- wape
|
| 87 |
+
- ed
|
| 88 |
+
- crps
|
| 89 |
+
- acd
|
| 90 |
+
- sd
|
| 91 |
+
- kd
|
| 92 |
+
- mdd
|
| 93 |
+
- fid
|
| 94 |
+
- sfd
|
| 95 |
+
- jftsd
|
| 96 |
+
- prdc_f1
|
| 97 |
+
- joint_prdc_f1
|
| 98 |
+
- cttp
|
| 99 |
+
- disc_auc
|
| 100 |
+
batch_size: 128
|
| 101 |
+
save_samples: true
|
| 102 |
+
sampler: ddim
|
| 103 |
+
display_interval: 10
|
| 104 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 105 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 106 |
+
cache_folder: __CACHE_ROOT__/timeweaver/synth-u/seed2
|
| 107 |
+
use_longalign: false
|
| 108 |
+
reference_split: train
|
| 109 |
+
disc_auc:
|
| 110 |
+
k_folds: 5
|
| 111 |
+
epochs: 5
|
| 112 |
+
batch_size: 128
|
| 113 |
+
lr: 0.001
|
| 114 |
+
weight_decay: 0.0001
|
| 115 |
+
hidden_channels: 128
|
| 116 |
+
num_layers: 3
|
| 117 |
+
dropout: 0.1
|
| 118 |
+
seed: 42
|
| 119 |
+
num_workers: 0
|
| 120 |
+
prdc:
|
| 121 |
+
k: 5
|
| 122 |
+
max_samples: 5000
|
| 123 |
+
seed: 0
|
| 124 |
+
metric: euclidean
|
| 125 |
+
backend: auto
|
| 126 |
+
knn_chunk_size: 4096
|
| 127 |
+
joint_enable: true
|
| 128 |
+
joint_weights:
|
| 129 |
+
ts: 1.0
|
| 130 |
+
text: 1.0
|
| 131 |
+
joint_normalize: standard
|
| 132 |
+
segment_classifier:
|
| 133 |
+
enable: false
|
| 134 |
+
segment_len: 128
|
| 135 |
+
n_segments: 3
|
| 136 |
+
seed: 42
|
| 137 |
+
acd_max_lag: 50
|
| 138 |
+
mdd_bins: 32
|
| 139 |
+
use_cache: false
|
| 140 |
+
cache_file: predictions_cache.pkl
|
| 141 |
+
viz:
|
| 142 |
+
enable: true
|
| 143 |
+
k_cases: 20
|
| 144 |
+
max_vars: 8
|
| 145 |
+
ncols: 5
|
| 146 |
+
seed: 42
|
| 147 |
+
dpi: 200
|
| 148 |
+
alpha: 0.25
|
| 149 |
+
figsize_per_subplot:
|
| 150 |
+
- 3.0
|
| 151 |
+
- 2.0
|
| 152 |
+
output_type: pdf
|
| 153 |
+
output_dir: ''
|
| 154 |
+
created_at: '2025-12-27T20:52:52.815536'
|
experiments/timeweaver/synth-u/seed2/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 6.306745138257389,
|
| 3 |
+
"wape": 102.75260034616316,
|
| 4 |
+
"ed": 10.294045974731445,
|
| 5 |
+
"crps": 0.5163008471568011,
|
| 6 |
+
"acd": 0.07299449294805527,
|
| 7 |
+
"sd": 0.0677454488779017,
|
| 8 |
+
"kd": 0.4350996962854301,
|
| 9 |
+
"mdd": 0.02103812662760416,
|
| 10 |
+
"fid": 52.15541755155982,
|
| 11 |
+
"sfd": 2.155792299061325,
|
| 12 |
+
"jftsd": 62.83588969110491,
|
| 13 |
+
"prdc_f1.f1": 0.20045689655172413,
|
| 14 |
+
"prdc_f1.precision": 0.58975,
|
| 15 |
+
"prdc_f1.recall": 0.12075,
|
| 16 |
+
"joint_prdc_f1.f1": 0.48181204627949187,
|
| 17 |
+
"joint_prdc_f1.precision": 0.74625,
|
| 18 |
+
"joint_prdc_f1.recall": 0.35575,
|
| 19 |
+
"cttp": 19.44007553100586,
|
| 20 |
+
"disc_auc": 0.9809070312500001
|
| 21 |
+
}
|
experiments/timeweaver/synth-u/seed2/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T21:10:21.273633",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/verbalts/synth-m/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_verbalts
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: verbalts
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
condition_type: adaLN
|
| 47 |
+
L_patch_len: 3
|
| 48 |
+
multipatch_num: 3
|
| 49 |
+
base_patch: 4
|
| 50 |
+
diffusion_steps: 50
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
side_var_emb: 16
|
| 56 |
+
side_time_emb: 112
|
| 57 |
+
var_dep_type: DEP
|
| 58 |
+
condition:
|
| 59 |
+
text:
|
| 60 |
+
enabled: true
|
| 61 |
+
input_dim: 1024
|
| 62 |
+
embedding_key: cap_emb
|
| 63 |
+
dropout: 0.0
|
| 64 |
+
text_projector: var_scale_diffstep_multi
|
| 65 |
+
num_stages: 3
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: false
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs: []
|
| 71 |
+
output_dim: 128
|
| 72 |
+
dropout: 0.0
|
| 73 |
+
label:
|
| 74 |
+
enabled: false
|
| 75 |
+
num_classes: 10
|
| 76 |
+
output_dim: 64
|
| 77 |
+
fusion: concat
|
| 78 |
+
condition_dropout: 0.0
|
| 79 |
+
eval:
|
| 80 |
+
n_samples: 10
|
| 81 |
+
metrics:
|
| 82 |
+
- dtw
|
| 83 |
+
- wape
|
| 84 |
+
- ed
|
| 85 |
+
- crps
|
| 86 |
+
- acd
|
| 87 |
+
- sd
|
| 88 |
+
- kd
|
| 89 |
+
- mdd
|
| 90 |
+
- fid
|
| 91 |
+
- sfd
|
| 92 |
+
- jftsd
|
| 93 |
+
- prdc_f1
|
| 94 |
+
- joint_prdc_f1
|
| 95 |
+
- cttp
|
| 96 |
+
- disc_auc
|
| 97 |
+
batch_size: 128
|
| 98 |
+
save_samples: true
|
| 99 |
+
sampler: ddim
|
| 100 |
+
display_interval: 10
|
| 101 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 102 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 103 |
+
cache_folder: __CACHE_ROOT__/verbalts/synth-m/seed0
|
| 104 |
+
use_longalign: false
|
| 105 |
+
reference_split: train
|
| 106 |
+
disc_auc:
|
| 107 |
+
k_folds: 5
|
| 108 |
+
epochs: 5
|
| 109 |
+
batch_size: 128
|
| 110 |
+
lr: 0.001
|
| 111 |
+
weight_decay: 0.0001
|
| 112 |
+
hidden_channels: 128
|
| 113 |
+
num_layers: 3
|
| 114 |
+
dropout: 0.1
|
| 115 |
+
seed: 42
|
| 116 |
+
num_workers: 0
|
| 117 |
+
prdc:
|
| 118 |
+
k: 5
|
| 119 |
+
max_samples: 5000
|
| 120 |
+
seed: 0
|
| 121 |
+
metric: euclidean
|
| 122 |
+
backend: auto
|
| 123 |
+
knn_chunk_size: 4096
|
| 124 |
+
joint_enable: true
|
| 125 |
+
joint_weights:
|
| 126 |
+
ts: 1.0
|
| 127 |
+
text: 1.0
|
| 128 |
+
joint_normalize: standard
|
| 129 |
+
segment_classifier:
|
| 130 |
+
enable: false
|
| 131 |
+
segment_len: 128
|
| 132 |
+
n_segments: 3
|
| 133 |
+
seed: 42
|
| 134 |
+
acd_max_lag: 50
|
| 135 |
+
mdd_bins: 32
|
| 136 |
+
use_cache: false
|
| 137 |
+
cache_file: predictions_cache.pkl
|
| 138 |
+
viz:
|
| 139 |
+
enable: true
|
| 140 |
+
k_cases: 20
|
| 141 |
+
max_vars: 8
|
| 142 |
+
ncols: 5
|
| 143 |
+
seed: 42
|
| 144 |
+
dpi: 200
|
| 145 |
+
alpha: 0.25
|
| 146 |
+
figsize_per_subplot:
|
| 147 |
+
- 3.0
|
| 148 |
+
- 2.0
|
| 149 |
+
output_type: pdf
|
| 150 |
+
output_dir: ''
|
| 151 |
+
created_at: '2025-12-27T18:27:33.408298'
|
experiments/verbalts/synth-m/seed0/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.908606059173055,
|
| 3 |
+
"wape": 88.21088966857347,
|
| 4 |
+
"ed": 12.982225296020507,
|
| 5 |
+
"crps": 0.44765225635860156,
|
| 6 |
+
"acd": 0.05096884444355965,
|
| 7 |
+
"sd": 0.03132322065263225,
|
| 8 |
+
"kd": 0.2212235214124354,
|
| 9 |
+
"mdd": 0.015057739257812494,
|
| 10 |
+
"fid": 30.656134730475003,
|
| 11 |
+
"sfd": 28.636071019387487,
|
| 12 |
+
"jftsd": 35.34921438451477,
|
| 13 |
+
"prdc_f1.f1": 0.4473662168450442,
|
| 14 |
+
"prdc_f1.precision": 0.757,
|
| 15 |
+
"prdc_f1.recall": 0.3175,
|
| 16 |
+
"joint_prdc_f1.f1": 0.6324860684089162,
|
| 17 |
+
"joint_prdc_f1.precision": 0.75875,
|
| 18 |
+
"joint_prdc_f1.recall": 0.54225,
|
| 19 |
+
"cttp": 24.750399078369142,
|
| 20 |
+
"disc_auc": 0.9902834375000001
|
| 21 |
+
}
|
experiments/verbalts/synth-m/seed0/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T20:02:44.214161",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/verbalts/synth-m/seed1/config.template.yaml
ADDED
|
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_verbalts
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 1
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: verbalts
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
condition_type: adaLN
|
| 47 |
+
L_patch_len: 3
|
| 48 |
+
multipatch_num: 3
|
| 49 |
+
base_patch: 4
|
| 50 |
+
diffusion_steps: 50
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
side_var_emb: 16
|
| 56 |
+
side_time_emb: 112
|
| 57 |
+
var_dep_type: DEP
|
| 58 |
+
condition:
|
| 59 |
+
text:
|
| 60 |
+
enabled: true
|
| 61 |
+
input_dim: 1024
|
| 62 |
+
embedding_key: cap_emb
|
| 63 |
+
dropout: 0.0
|
| 64 |
+
text_projector: var_scale_diffstep_multi
|
| 65 |
+
num_stages: 3
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: false
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs: []
|
| 71 |
+
output_dim: 128
|
| 72 |
+
dropout: 0.0
|
| 73 |
+
label:
|
| 74 |
+
enabled: false
|
| 75 |
+
num_classes: 10
|
| 76 |
+
output_dim: 64
|
| 77 |
+
fusion: concat
|
| 78 |
+
condition_dropout: 0.0
|
| 79 |
+
eval:
|
| 80 |
+
n_samples: 10
|
| 81 |
+
metrics:
|
| 82 |
+
- dtw
|
| 83 |
+
- wape
|
| 84 |
+
- ed
|
| 85 |
+
- crps
|
| 86 |
+
- acd
|
| 87 |
+
- sd
|
| 88 |
+
- kd
|
| 89 |
+
- mdd
|
| 90 |
+
- fid
|
| 91 |
+
- sfd
|
| 92 |
+
- jftsd
|
| 93 |
+
- prdc_f1
|
| 94 |
+
- joint_prdc_f1
|
| 95 |
+
- cttp
|
| 96 |
+
- disc_auc
|
| 97 |
+
batch_size: 128
|
| 98 |
+
save_samples: true
|
| 99 |
+
sampler: ddim
|
| 100 |
+
display_interval: 10
|
| 101 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 102 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 103 |
+
cache_folder: __CACHE_ROOT__/verbalts/synth-m/seed1
|
| 104 |
+
use_longalign: false
|
| 105 |
+
reference_split: train
|
| 106 |
+
disc_auc:
|
| 107 |
+
k_folds: 5
|
| 108 |
+
epochs: 5
|
| 109 |
+
batch_size: 128
|
| 110 |
+
lr: 0.001
|
| 111 |
+
weight_decay: 0.0001
|
| 112 |
+
hidden_channels: 128
|
| 113 |
+
num_layers: 3
|
| 114 |
+
dropout: 0.1
|
| 115 |
+
seed: 42
|
| 116 |
+
num_workers: 0
|
| 117 |
+
prdc:
|
| 118 |
+
k: 5
|
| 119 |
+
max_samples: 5000
|
| 120 |
+
seed: 0
|
| 121 |
+
metric: euclidean
|
| 122 |
+
backend: auto
|
| 123 |
+
knn_chunk_size: 4096
|
| 124 |
+
joint_enable: true
|
| 125 |
+
joint_weights:
|
| 126 |
+
ts: 1.0
|
| 127 |
+
text: 1.0
|
| 128 |
+
joint_normalize: standard
|
| 129 |
+
segment_classifier:
|
| 130 |
+
enable: false
|
| 131 |
+
segment_len: 128
|
| 132 |
+
n_segments: 3
|
| 133 |
+
seed: 42
|
| 134 |
+
acd_max_lag: 50
|
| 135 |
+
mdd_bins: 32
|
| 136 |
+
use_cache: false
|
| 137 |
+
cache_file: predictions_cache.pkl
|
| 138 |
+
viz:
|
| 139 |
+
enable: true
|
| 140 |
+
k_cases: 20
|
| 141 |
+
max_vars: 8
|
| 142 |
+
ncols: 5
|
| 143 |
+
seed: 42
|
| 144 |
+
dpi: 200
|
| 145 |
+
alpha: 0.25
|
| 146 |
+
figsize_per_subplot:
|
| 147 |
+
- 3.0
|
| 148 |
+
- 2.0
|
| 149 |
+
output_type: pdf
|
| 150 |
+
output_dir: ''
|
| 151 |
+
created_at: '2025-12-27T18:25:58.276238'
|
experiments/verbalts/synth-m/seed1/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 9.688534115960435,
|
| 3 |
+
"wape": 86.73450378196509,
|
| 4 |
+
"ed": 12.808458847045898,
|
| 5 |
+
"crps": 0.44154393791529506,
|
| 6 |
+
"acd": 0.04938828572630882,
|
| 7 |
+
"sd": 0.013795681588651848,
|
| 8 |
+
"kd": 0.23307523080260983,
|
| 9 |
+
"mdd": 0.015345753987630202,
|
| 10 |
+
"fid": 30.426586720716642,
|
| 11 |
+
"sfd": 28.257033437889987,
|
| 12 |
+
"jftsd": 34.652391105447236,
|
| 13 |
+
"prdc_f1.f1": 0.4396665895953757,
|
| 14 |
+
"prdc_f1.precision": 0.77425,
|
| 15 |
+
"prdc_f1.recall": 0.307,
|
| 16 |
+
"joint_prdc_f1.f1": 0.6421972416399018,
|
| 17 |
+
"joint_prdc_f1.precision": 0.775,
|
| 18 |
+
"joint_prdc_f1.recall": 0.54825,
|
| 19 |
+
"cttp": 25.439562438964845,
|
| 20 |
+
"disc_auc": 0.9904329687499999
|
| 21 |
+
}
|
experiments/verbalts/synth-m/seed1/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T20:30:55.671010",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/verbalts/synth-m/seed2/config.template.yaml
ADDED
|
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-m_verbalts
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 2
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-m
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-m
|
| 35 |
+
n_var: 2
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: verbalts
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
condition_type: adaLN
|
| 47 |
+
L_patch_len: 3
|
| 48 |
+
multipatch_num: 3
|
| 49 |
+
base_patch: 4
|
| 50 |
+
diffusion_steps: 50
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
side_var_emb: 16
|
| 56 |
+
side_time_emb: 112
|
| 57 |
+
var_dep_type: DEP
|
| 58 |
+
condition:
|
| 59 |
+
text:
|
| 60 |
+
enabled: true
|
| 61 |
+
input_dim: 1024
|
| 62 |
+
embedding_key: cap_emb
|
| 63 |
+
dropout: 0.0
|
| 64 |
+
text_projector: var_scale_diffstep_multi
|
| 65 |
+
num_stages: 3
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: false
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs: []
|
| 71 |
+
output_dim: 128
|
| 72 |
+
dropout: 0.0
|
| 73 |
+
label:
|
| 74 |
+
enabled: false
|
| 75 |
+
num_classes: 10
|
| 76 |
+
output_dim: 64
|
| 77 |
+
fusion: concat
|
| 78 |
+
condition_dropout: 0.0
|
| 79 |
+
eval:
|
| 80 |
+
n_samples: 10
|
| 81 |
+
metrics:
|
| 82 |
+
- dtw
|
| 83 |
+
- wape
|
| 84 |
+
- ed
|
| 85 |
+
- crps
|
| 86 |
+
- acd
|
| 87 |
+
- sd
|
| 88 |
+
- kd
|
| 89 |
+
- mdd
|
| 90 |
+
- fid
|
| 91 |
+
- sfd
|
| 92 |
+
- jftsd
|
| 93 |
+
- prdc_f1
|
| 94 |
+
- joint_prdc_f1
|
| 95 |
+
- cttp
|
| 96 |
+
- disc_auc
|
| 97 |
+
batch_size: 128
|
| 98 |
+
save_samples: true
|
| 99 |
+
sampler: ddim
|
| 100 |
+
display_interval: 10
|
| 101 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-m/model_configs.yaml
|
| 102 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-m/clip_model_best.pth
|
| 103 |
+
cache_folder: __CACHE_ROOT__/verbalts/synth-m/seed2
|
| 104 |
+
use_longalign: false
|
| 105 |
+
reference_split: train
|
| 106 |
+
disc_auc:
|
| 107 |
+
k_folds: 5
|
| 108 |
+
epochs: 5
|
| 109 |
+
batch_size: 128
|
| 110 |
+
lr: 0.001
|
| 111 |
+
weight_decay: 0.0001
|
| 112 |
+
hidden_channels: 128
|
| 113 |
+
num_layers: 3
|
| 114 |
+
dropout: 0.1
|
| 115 |
+
seed: 42
|
| 116 |
+
num_workers: 0
|
| 117 |
+
prdc:
|
| 118 |
+
k: 5
|
| 119 |
+
max_samples: 5000
|
| 120 |
+
seed: 0
|
| 121 |
+
metric: euclidean
|
| 122 |
+
backend: auto
|
| 123 |
+
knn_chunk_size: 4096
|
| 124 |
+
joint_enable: true
|
| 125 |
+
joint_weights:
|
| 126 |
+
ts: 1.0
|
| 127 |
+
text: 1.0
|
| 128 |
+
joint_normalize: standard
|
| 129 |
+
segment_classifier:
|
| 130 |
+
enable: false
|
| 131 |
+
segment_len: 128
|
| 132 |
+
n_segments: 3
|
| 133 |
+
seed: 42
|
| 134 |
+
acd_max_lag: 50
|
| 135 |
+
mdd_bins: 32
|
| 136 |
+
use_cache: false
|
| 137 |
+
cache_file: predictions_cache.pkl
|
| 138 |
+
viz:
|
| 139 |
+
enable: true
|
| 140 |
+
k_cases: 20
|
| 141 |
+
max_vars: 8
|
| 142 |
+
ncols: 5
|
| 143 |
+
seed: 42
|
| 144 |
+
dpi: 200
|
| 145 |
+
alpha: 0.25
|
| 146 |
+
figsize_per_subplot:
|
| 147 |
+
- 3.0
|
| 148 |
+
- 2.0
|
| 149 |
+
output_type: pdf
|
| 150 |
+
output_dir: ''
|
| 151 |
+
created_at: '2025-12-27T18:25:56.863507'
|
experiments/verbalts/synth-m/seed2/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 10.922539546646847,
|
| 3 |
+
"wape": 95.35738398535062,
|
| 4 |
+
"ed": 13.85847917175293,
|
| 5 |
+
"crps": 0.5405318572121146,
|
| 6 |
+
"acd": 0.053873561322689056,
|
| 7 |
+
"sd": 0.07642871427613498,
|
| 8 |
+
"kd": 0.1231948631156774,
|
| 9 |
+
"mdd": 0.015523325602213537,
|
| 10 |
+
"fid": 39.61011326177959,
|
| 11 |
+
"sfd": 33.06918106665158,
|
| 12 |
+
"jftsd": 44.09657725747664,
|
| 13 |
+
"prdc_f1.f1": 0.3634620858895706,
|
| 14 |
+
"prdc_f1.precision": 0.782,
|
| 15 |
+
"prdc_f1.recall": 0.23675,
|
| 16 |
+
"joint_prdc_f1.f1": 0.528560749724366,
|
| 17 |
+
"joint_prdc_f1.precision": 0.71425,
|
| 18 |
+
"joint_prdc_f1.recall": 0.4195,
|
| 19 |
+
"cttp": 23.793466705322267,
|
| 20 |
+
"disc_auc": 0.993119375
|
| 21 |
+
}
|
experiments/verbalts/synth-m/seed2/summary.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"status": "completed",
|
| 3 |
+
"finished_at": "2025-12-27T19:40:18.150148",
|
| 4 |
+
"best_checkpoint": "checkpoints/finetune/best.ckpt"
|
| 5 |
+
}
|
experiments/verbalts/synth-u/seed0/config.template.yaml
ADDED
|
@@ -0,0 +1,151 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: synth-u_verbalts
|
| 2 |
+
description: ''
|
| 3 |
+
seed: 0
|
| 4 |
+
device: cuda:0
|
| 5 |
+
train:
|
| 6 |
+
stages:
|
| 7 |
+
- name: finetune
|
| 8 |
+
epochs: 700
|
| 9 |
+
lr: 0.001
|
| 10 |
+
use_condition: true
|
| 11 |
+
freeze_modules: []
|
| 12 |
+
early_stopping_patience: 50
|
| 13 |
+
epochs: 700
|
| 14 |
+
batch_size: 256
|
| 15 |
+
lr: 0.001
|
| 16 |
+
weight_decay: 0.0001
|
| 17 |
+
scheduler: cosine
|
| 18 |
+
scheduler_params: {}
|
| 19 |
+
early_stopping_patience: 50
|
| 20 |
+
gradient_clip_val: 1.0
|
| 21 |
+
accumulate_grad_batches: 1
|
| 22 |
+
val_check_interval: 1.0
|
| 23 |
+
num_workers: 4
|
| 24 |
+
pin_memory: true
|
| 25 |
+
limit_train_batches: 1.0
|
| 26 |
+
limit_val_batches: 1.0
|
| 27 |
+
limit_test_batches: 1.0
|
| 28 |
+
num_sanity_val_steps: 2
|
| 29 |
+
log_grad_norm: true
|
| 30 |
+
log_param_norm: true
|
| 31 |
+
log_norm_every_n_steps: 50
|
| 32 |
+
data:
|
| 33 |
+
name: synth-u
|
| 34 |
+
data_folder: __DATA_ROOT__/datasets/synth-u
|
| 35 |
+
n_var: 1
|
| 36 |
+
seq_length: 128
|
| 37 |
+
normalize: false
|
| 38 |
+
train_split: 0.8
|
| 39 |
+
val_split: 0.1
|
| 40 |
+
model:
|
| 41 |
+
name: verbalts
|
| 42 |
+
channels: 64
|
| 43 |
+
layers: 3
|
| 44 |
+
nheads: 8
|
| 45 |
+
dropout: 0.1
|
| 46 |
+
condition_type: adaLN
|
| 47 |
+
L_patch_len: 3
|
| 48 |
+
multipatch_num: 3
|
| 49 |
+
base_patch: 4
|
| 50 |
+
diffusion_steps: 50
|
| 51 |
+
attention_mask_type: parallel
|
| 52 |
+
noise_schedule: quad
|
| 53 |
+
beta_start: 0.0001
|
| 54 |
+
beta_end: 0.5
|
| 55 |
+
side_var_emb: 16
|
| 56 |
+
side_time_emb: 112
|
| 57 |
+
var_dep_type: DEP
|
| 58 |
+
condition:
|
| 59 |
+
text:
|
| 60 |
+
enabled: true
|
| 61 |
+
input_dim: 1024
|
| 62 |
+
embedding_key: cap_emb
|
| 63 |
+
dropout: 0.0
|
| 64 |
+
text_projector: var_scale_diffstep_multi
|
| 65 |
+
num_stages: 3
|
| 66 |
+
cfg_scale: 1.0
|
| 67 |
+
attribute:
|
| 68 |
+
enabled: false
|
| 69 |
+
continuous_dim: 0
|
| 70 |
+
discrete_configs: []
|
| 71 |
+
output_dim: 128
|
| 72 |
+
dropout: 0.0
|
| 73 |
+
label:
|
| 74 |
+
enabled: false
|
| 75 |
+
num_classes: 10
|
| 76 |
+
output_dim: 64
|
| 77 |
+
fusion: concat
|
| 78 |
+
condition_dropout: 0.0
|
| 79 |
+
eval:
|
| 80 |
+
n_samples: 10
|
| 81 |
+
metrics:
|
| 82 |
+
- dtw
|
| 83 |
+
- wape
|
| 84 |
+
- ed
|
| 85 |
+
- crps
|
| 86 |
+
- acd
|
| 87 |
+
- sd
|
| 88 |
+
- kd
|
| 89 |
+
- mdd
|
| 90 |
+
- fid
|
| 91 |
+
- sfd
|
| 92 |
+
- jftsd
|
| 93 |
+
- prdc_f1
|
| 94 |
+
- joint_prdc_f1
|
| 95 |
+
- cttp
|
| 96 |
+
- disc_auc
|
| 97 |
+
batch_size: 128
|
| 98 |
+
save_samples: true
|
| 99 |
+
sampler: ddim
|
| 100 |
+
display_interval: 10
|
| 101 |
+
clip_config_path: __RELEASE_ROOT__/resources/cttp/synth-u/model_configs.yaml
|
| 102 |
+
clip_model_path: __RELEASE_ROOT__/resources/cttp/synth-u/clip_model_best.pth
|
| 103 |
+
cache_folder: __CACHE_ROOT__/verbalts/synth-u/seed0
|
| 104 |
+
use_longalign: false
|
| 105 |
+
reference_split: train
|
| 106 |
+
disc_auc:
|
| 107 |
+
k_folds: 5
|
| 108 |
+
epochs: 5
|
| 109 |
+
batch_size: 128
|
| 110 |
+
lr: 0.001
|
| 111 |
+
weight_decay: 0.0001
|
| 112 |
+
hidden_channels: 128
|
| 113 |
+
num_layers: 3
|
| 114 |
+
dropout: 0.1
|
| 115 |
+
seed: 42
|
| 116 |
+
num_workers: 0
|
| 117 |
+
prdc:
|
| 118 |
+
k: 5
|
| 119 |
+
max_samples: 5000
|
| 120 |
+
seed: 0
|
| 121 |
+
metric: euclidean
|
| 122 |
+
backend: auto
|
| 123 |
+
knn_chunk_size: 4096
|
| 124 |
+
joint_enable: true
|
| 125 |
+
joint_weights:
|
| 126 |
+
ts: 1.0
|
| 127 |
+
text: 1.0
|
| 128 |
+
joint_normalize: standard
|
| 129 |
+
segment_classifier:
|
| 130 |
+
enable: false
|
| 131 |
+
segment_len: 128
|
| 132 |
+
n_segments: 3
|
| 133 |
+
seed: 42
|
| 134 |
+
acd_max_lag: 50
|
| 135 |
+
mdd_bins: 32
|
| 136 |
+
use_cache: false
|
| 137 |
+
cache_file: predictions_cache.pkl
|
| 138 |
+
viz:
|
| 139 |
+
enable: true
|
| 140 |
+
k_cases: 20
|
| 141 |
+
max_vars: 8
|
| 142 |
+
ncols: 5
|
| 143 |
+
seed: 42
|
| 144 |
+
dpi: 200
|
| 145 |
+
alpha: 0.25
|
| 146 |
+
figsize_per_subplot:
|
| 147 |
+
- 3.0
|
| 148 |
+
- 2.0
|
| 149 |
+
output_type: pdf
|
| 150 |
+
output_dir: ''
|
| 151 |
+
created_at: '2025-12-27T18:28:14.826759'
|
experiments/verbalts/synth-u/seed0/results/expected_seed_metrics.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dtw": 4.930648105329671,
|
| 3 |
+
"wape": 83.84086095919642,
|
| 4 |
+
"ed": 8.562691368103028,
|
| 5 |
+
"crps": 0.44792016319441924,
|
| 6 |
+
"acd": 0.0560174435377121,
|
| 7 |
+
"sd": 0.0032578879594434736,
|
| 8 |
+
"kd": 0.3045949253759903,
|
| 9 |
+
"mdd": 0.013916402180989579,
|
| 10 |
+
"fid": 38.01059001158677,
|
| 11 |
+
"sfd": 2.035234186940542,
|
| 12 |
+
"jftsd": 41.473615746430596,
|
| 13 |
+
"prdc_f1.f1": 0.3203814049586777,
|
| 14 |
+
"prdc_f1.precision": 0.69975,
|
| 15 |
+
"prdc_f1.recall": 0.20775,
|
| 16 |
+
"joint_prdc_f1.f1": 0.8099794375566979,
|
| 17 |
+
"joint_prdc_f1.precision": 0.9445,
|
| 18 |
+
"joint_prdc_f1.recall": 0.709,
|
| 19 |
+
"cttp": 26.90989910888672,
|
| 20 |
+
"disc_auc": 0.9774965625
|
| 21 |
+
}
|