diversion26 / model_config.json
agentsay's picture
Upload 6 files
b9abc4e verified
raw
history blame contribute delete
752 Bytes
{
"n_features": 14,
"n_physics": 3,
"hidden": 32,
"n_heads": 2,
"dropout": 0.4,
"seq_len": 8,
"horizon": 2,
"feature_cols": [
"lat",
"lon",
"cumulative_disp_mm",
"incremental_disp_mm",
"velocity_last_3ep",
"velocity_last_6ep",
"disp_rolling_mean_3",
"disp_rolling_mean_6",
"disp_rolling_std_6",
"coherence",
"mean_velocity_mm_yr",
"dem_height_m",
"seasonal_sin",
"seasonal_cos"
],
"physics_cols": [
"acceleration",
"velocity_last_3ep",
"dem_height_m"
],
"best_threshold": 0.5551024079322815,
"best_val_auc": 0.8073059156530157,
"test_auc": 0.8027481911831947,
"test_pr_auc": 0.2522120332201386,
"pos_weight": 11.660771704180064,
"stopped_epoch": 36
}