remove sub prevalene

prevalence/preprocessing_config.json

@@ -20,6 +20,7 @@
     "lsm",
     "routine"
   ],
+  "window_size": 14,
   "intervention_day": 3285,
   "predictor": "prevalence",
   "eps_prevalence": 1e-05,

prevalence/prevalence/checkpoint/0/_CHECKPOINT_METADATA

@@ -1 +0,0 @@
-{"item_handlers": {"model": "orbax.checkpoint.experimental.v1._src.handlers.pytree_handler.PyTreeHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1781690568898697676, "commit_timestamp_nsecs": 1781690569030953030, "custom_metadata": {}}

prevalence/prevalence/checkpoint/0/model/_METADATA

@@ -1 +0,0 @@
-{"tree_metadata": {"('dropout', 'rngs', 'count', 'value')": {"key_metadata": [{"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "count", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": []}}, "('dropout', 'rngs', 'key', 'value')": {"key_metadata": [{"key": "dropout", "key_type": 2}, {"key": "rngs", "key_type": 2}, {"key": "key", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('input_proj', 'bias', 'value')": {"key_metadata": [{"key": "input_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('input_proj', 'kernel', 'value')": {"key_metadata": [{"key": "input_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16, 256]}}, "('mamba2', 'embedder', 'embedding', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "embedder", "key_type": 2}, {"key": "embedding", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 256]}}, "('mamba2', 'final_norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "final_norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('mamba2', 'layers', '0', 'mixer', 'A_log', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "A_log", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '0', 'mixer', 'D', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "D", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '0', 'mixer', 'conv1d', 'conv', 'bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [768]}}, "('mamba2', 'layers', '0', 'mixer', 'conv1d', 'conv', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 1, 768]}}, "('mamba2', 'layers', '0', 'mixer', 'dt_bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "dt_bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '0', 'mixer', 'in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1288]}}, "('mamba2', 'layers', '0', 'mixer', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('mamba2', 'layers', '0', 'mixer', 'out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 256]}}, "('mamba2', 'layers', '0', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('mamba2', 'layers', '1', 'mixer', 'A_log', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "A_log", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '1', 'mixer', 'D', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "D", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '1', 'mixer', 'conv1d', 'conv', 'bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [768]}}, "('mamba2', 'layers', '1', 'mixer', 'conv1d', 'conv', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 1, 768]}}, "('mamba2', 'layers', '1', 'mixer', 'dt_bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "dt_bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '1', 'mixer', 'in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1288]}}, "('mamba2', 'layers', '1', 'mixer', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('mamba2', 'layers', '1', 'mixer', 'out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 256]}}, "('mamba2', 'layers', '1', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('mamba2', 'layers', '2', 'mixer', 'A_log', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "A_log", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '2', 'mixer', 'D', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "D", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '2', 'mixer', 'conv1d', 'conv', 'bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [768]}}, "('mamba2', 'layers', '2', 'mixer', 'conv1d', 'conv', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 1, 768]}}, "('mamba2', 'layers', '2', 'mixer', 'dt_bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "dt_bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '2', 'mixer', 'in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1288]}}, "('mamba2', 'layers', '2', 'mixer', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('mamba2', 'layers', '2', 'mixer', 'out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 256]}}, "('mamba2', 'layers', '2', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('mamba2', 'layers', '3', 'mixer', 'A_log', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "A_log", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '3', 'mixer', 'D', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "D", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '3', 'mixer', 'conv1d', 'conv', 'bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [768]}}, "('mamba2', 'layers', '3', 'mixer', 'conv1d', 'conv', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "conv1d", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4, 1, 768]}}, "('mamba2', 'layers', '3', 'mixer', 'dt_bias', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "dt_bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [8]}}, "('mamba2', 'layers', '3', 'mixer', 'in_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "in_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1288]}}, "('mamba2', 'layers', '3', 'mixer', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('mamba2', 'layers', '3', 'mixer', 'out_proj', 'kernel', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "mixer", "key_type": 2}, {"key": "out_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 256]}}, "('mamba2', 'layers', '3', 'norm', 'weight', 'value')": {"key_metadata": [{"key": "mamba2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "norm", "key_type": 2}, {"key": "weight", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('output_proj', 'bias', 'value')": {"key_metadata": [{"key": "output_proj", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1]}}, "('output_proj', 'kernel', 'value')": {"key_metadata": [{"key": "output_proj", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 1]}}}, "use_ocdbt": true, "use_zarr3": true, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

prevalence/prevalence/checkpoint/0/model/_sharding

@@ -1 +0,0 @@
-{"ZHJvcG91dC5ybmdzLmNvdW50LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZHJvcG91dC5ybmdzLmtleS52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","aW5wdXRfcHJvai5iaWFzLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","aW5wdXRfcHJvai5rZXJuZWwudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","b3V0cHV0X3Byb2ouYmlhcy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","b3V0cHV0X3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmVtYmVkZGVyLmVtYmVkZGluZy52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmZpbmFsX25vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLkFfbG9nLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLkQudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLm91dF9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLmNvbnYxZC5jb252LmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLmNvbnYxZC5jb252Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLmR0X2JpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm1peGVyLmluX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4wLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLkFfbG9nLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLkQudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLm91dF9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLmNvbnYxZC5jb252LmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLmNvbnYxZC5jb252Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLmR0X2JpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm1peGVyLmluX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4xLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLkFfbG9nLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLkQudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLm91dF9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLmNvbnYxZC5jb252LmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLmNvbnYxZC5jb252Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLmR0X2JpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm1peGVyLmluX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4yLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLkFfbG9nLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLkQudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLm91dF9wcm9qLmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLmNvbnYxZC5jb252LmJpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLmNvbnYxZC5jb252Lmtlcm5lbC52YWx1ZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLmR0X2JpYXMudmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm1peGVyLmluX3Byb2oua2VybmVsLnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bWFtYmEyLmxheWVycy4zLm5vcm0ud2VpZ2h0LnZhbHVl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

prevalence/prevalence/checkpoint/0/model/array_metadatas/process_0

@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "dropout.rngs.count.value", "write_shape": [], "chunk_shape": [], "ext_metadata": null}}, {"array_metadata": {"param_name": "dropout.rngs.key.value", "write_shape": [2], "chunk_shape": [2], "ext_metadata": {"random_key_impl": "threefry2x32"}}}, {"array_metadata": {"param_name": "input_proj.bias.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "input_proj.kernel.value", "write_shape": [16, 256], "chunk_shape": [16, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.embedder.embedding.value", "write_shape": [1, 256], "chunk_shape": [1, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.final_norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.A_log.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.D.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.conv1d.conv.bias.value", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.conv1d.conv.kernel.value", "write_shape": [4, 1, 768], "chunk_shape": [4, 1, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.dt_bias.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.in_proj.kernel.value", "write_shape": [256, 1288], "chunk_shape": [256, 1288], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.norm.weight.value", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.mixer.out_proj.kernel.value", "write_shape": [512, 256], "chunk_shape": [512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.0.norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.A_log.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.D.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.conv1d.conv.bias.value", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.conv1d.conv.kernel.value", "write_shape": [4, 1, 768], "chunk_shape": [4, 1, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.dt_bias.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.in_proj.kernel.value", "write_shape": [256, 1288], "chunk_shape": [256, 1288], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.norm.weight.value", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.mixer.out_proj.kernel.value", "write_shape": [512, 256], "chunk_shape": [512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.1.norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.A_log.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.D.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.conv1d.conv.bias.value", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.conv1d.conv.kernel.value", "write_shape": [4, 1, 768], "chunk_shape": [4, 1, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.dt_bias.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.in_proj.kernel.value", "write_shape": [256, 1288], "chunk_shape": [256, 1288], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.norm.weight.value", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.mixer.out_proj.kernel.value", "write_shape": [512, 256], "chunk_shape": [512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.2.norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.A_log.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.D.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.conv1d.conv.bias.value", "write_shape": [768], "chunk_shape": [768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.conv1d.conv.kernel.value", "write_shape": [4, 1, 768], "chunk_shape": [4, 1, 768], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.dt_bias.value", "write_shape": [8], "chunk_shape": [8], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.in_proj.kernel.value", "write_shape": [256, 1288], "chunk_shape": [256, 1288], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.norm.weight.value", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.mixer.out_proj.kernel.value", "write_shape": [512, 256], "chunk_shape": [512, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "mamba2.layers.3.norm.weight.value", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_proj.bias.value", "write_shape": [1], "chunk_shape": [1], "ext_metadata": null}}, {"array_metadata": {"param_name": "output_proj.kernel.value", "write_shape": [256, 1], "chunk_shape": [256, 1], "ext_metadata": null}}]}

prevalence/prevalence/checkpoint/0/model/ocdbt.process_0/d/cfaf63ebed00aa0884aeab11dad58eec

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9e1a33012b23046d23e38ab02e0dad0e6d376b853871a910d53240a70c58ca0d
-size 6893568

prevalence/prevalence/model_config.json

@@ -1,15 +0,0 @@
-{
-  "model_type": "Mamba2Regressor",
-  "predictor": "prevalence",
-  "input_size": 16,
-  "d_model": 256,
-  "n_layers": 4,
-  "d_state": 128,
-  "d_conv": 4,
-  "expand": 2,
-  "head_dim": 64,
-  "chunk_size": 256,
-  "output_dim": 1,
-  "dropout": 0.3,
-  "seed": 42
-}

prevalence/prevalence/preprocessing_config.json

@@ -1,55 +0,0 @@
-{
-  "static_covars": [
-    "eir",
-    "dn0_use",
-    "dn0_future",
-    "Q0",
-    "phi_bednets",
-    "seasonal",
-    "routine",
-    "itn_use",
-    "irs_use",
-    "itn_future",
-    "irs_future",
-    "lsm"
-  ],
-  "after_intervention": [
-    "dn0_future",
-    "itn_future",
-    "irs_future",
-    "lsm",
-    "routine"
-  ],
-  "intervention_day": 3285,
-  "use_cyclical_time": true,
-  "predictor": "prevalence",
-  "eps_prevalence": 1e-05,
-  "scaler_mean": [
-    58.74008560180664,
-    0.31604108214378357,
-    0.3175233006477356,
-    0.7978994250297546,
-    0.6636720895767212,
-    0.482501357793808,
-    0.48787713050842285,
-    0.25194787979125977,
-    0.15634511411190033,
-    0.25312861800193787,
-    0.17558598518371582,
-    0.24548254907131195
-  ],
-  "scaler_scale": [
-    104.59117126464844,
-    0.16008581221103668,
-    0.15946531295776367,
-    0.11743186414241791,
-    0.16026772558689117,
-    0.49969369173049927,
-    0.4998529851436615,
-    0.32460200786590576,
-    0.26269596815109253,
-    0.32312172651290894,
-    0.2886512279510498,
-    0.32189667224884033
-  ]
-}