bella-companion 0.0.47__py3-none-any.whl → 0.1.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of bella-companion might be problematic. Click here for more details.

Files changed (83) hide show
  1. bella_companion/.DS_Store +0 -0
  2. bella_companion/eucovid/beast_configs/GLM.xml +28 -18
  3. bella_companion/eucovid/beast_configs/MLP.xml +3 -4
  4. bella_companion/eucovid/run_beast.py +13 -95
  5. bella_companion/features.py +13 -0
  6. bella_companion/simulations/figures/epi_multitype.py +5 -5
  7. bella_companion/simulations/figures/fbd_2traits.py +5 -5
  8. bella_companion/simulations/scenarios/epi_multitype.py +1 -1
  9. bella_companion/simulations/scenarios/epi_skyline.py +1 -1
  10. bella_companion/simulations/scenarios/fbd_2traits.py +1 -1
  11. bella_companion/simulations/scenarios/fbd_no_traits.py +1 -1
  12. bella_companion/simulations/scenarios/scenario.py +1 -1
  13. bella_companion/utils/__init__.py +4 -4
  14. bella_companion/utils/beast.py +2 -2
  15. bella_companion/utils/explain/__init__.py +7 -0
  16. bella_companion/{simulations/figures → utils}/explain/pdp.py +61 -18
  17. bella_companion/utils/explain/shap.py +104 -0
  18. bella_companion/utils/explain/typeguards.py +20 -0
  19. {bella_companion-0.0.47.dist-info → bella_companion-0.1.1.dist-info}/METADATA +8 -5
  20. bella_companion-0.1.1.dist-info/RECORD +64 -0
  21. bella_companion-0.1.1.dist-info/WHEEL +4 -0
  22. {bella_companion-0.0.47.dist-info → bella_companion-0.1.1.dist-info}/entry_points.txt +1 -0
  23. bella_companion/__pycache__/__init__.cpython-310.pyc +0 -0
  24. bella_companion/__pycache__/cli.cpython-310.pyc +0 -0
  25. bella_companion/eucovid/__pycache__/__init__.cpython-310.pyc +0 -0
  26. bella_companion/eucovid/__pycache__/run_beast.cpython-310.pyc +0 -0
  27. bella_companion/eucovid/beast_configs/Nonparametric.xml +0 -399
  28. bella_companion/eucovid/data/case_data.tsv +0 -346
  29. bella_companion/eucovid/data/change_dates.tsv +0 -8
  30. bella_companion/eucovid/data/osi.tsv +0 -376
  31. bella_companion/eucovid/data/predictors/all/closedborders_7e_nt.tsv +0 -35
  32. bella_companion/eucovid/data/predictors/all/dist_7e_ls.tsv +0 -35
  33. bella_companion/eucovid/data/predictors/all/flight_7e_ls.tsv +0 -35
  34. bella_companion/eucovid/data/predictors/all/flight_pop_x_7e_ls.tsv +0 -35
  35. bella_companion/eucovid/data/predictors/all/osi_x_7e_ls.tsv +0 -35
  36. bella_companion/eucovid/data/predictors/all/osi_y_7e_ls.tsv +0 -35
  37. bella_companion/eucovid/data/predictors/all/pop_x_7e_ls.tsv +0 -35
  38. bella_companion/eucovid/data/predictors/all/pop_y_7e_ls.tsv +0 -35
  39. bella_companion/eucovid/data/predictors/all/sharedborders_7e_nt.tsv +0 -35
  40. bella_companion/eucovid/data/predictors/changetimes_all_7e.tsv +0 -6
  41. bella_companion/platyrrhine/__pycache__/__init__.cpython-310.pyc +0 -0
  42. bella_companion/platyrrhine/__pycache__/results.cpython-310.pyc +0 -0
  43. bella_companion/platyrrhine/__pycache__/run_beast.cpython-310.pyc +0 -0
  44. bella_companion/platyrrhine/__pycache__/summarize.cpython-310.pyc +0 -0
  45. bella_companion/platyrrhine/__pycache__/summarize_logs.cpython-310.pyc +0 -0
  46. bella_companion/simulations/__pycache__/__init__.cpython-310.pyc +0 -0
  47. bella_companion/simulations/__pycache__/features.cpython-310.pyc +0 -0
  48. bella_companion/simulations/__pycache__/generate_data.cpython-310.pyc +0 -0
  49. bella_companion/simulations/__pycache__/generate_figures.cpython-310.pyc +0 -0
  50. bella_companion/simulations/__pycache__/metrics.cpython-310.pyc +0 -0
  51. bella_companion/simulations/__pycache__/run_beast.cpython-310.pyc +0 -0
  52. bella_companion/simulations/__pycache__/summarize.cpython-310.pyc +0 -0
  53. bella_companion/simulations/__pycache__/summarize_logs.cpython-310.pyc +0 -0
  54. bella_companion/simulations/features.py +0 -7
  55. bella_companion/simulations/figures/__pycache__/__init__.cpython-310.pyc +0 -0
  56. bella_companion/simulations/figures/__pycache__/epi_multitype.cpython-310.pyc +0 -0
  57. bella_companion/simulations/figures/__pycache__/epi_skyline.cpython-310.pyc +0 -0
  58. bella_companion/simulations/figures/__pycache__/fbd_2traits.cpython-310.pyc +0 -0
  59. bella_companion/simulations/figures/__pycache__/fbd_no_traits.cpython-310.pyc +0 -0
  60. bella_companion/simulations/figures/__pycache__/scenarios.cpython-310.pyc +0 -0
  61. bella_companion/simulations/figures/__pycache__/utils.cpython-310.pyc +0 -0
  62. bella_companion/simulations/figures/explain/__init__.py +0 -6
  63. bella_companion/simulations/figures/explain/__pycache__/__init__.cpython-310.pyc +0 -0
  64. bella_companion/simulations/figures/explain/__pycache__/pdp.cpython-310.pyc +0 -0
  65. bella_companion/simulations/figures/explain/__pycache__/shap.cpython-310.pyc +0 -0
  66. bella_companion/simulations/figures/explain/shap.py +0 -62
  67. bella_companion/simulations/scenarios/__pycache__/__init__.cpython-310.pyc +0 -0
  68. bella_companion/simulations/scenarios/__pycache__/common.cpython-310.pyc +0 -0
  69. bella_companion/simulations/scenarios/__pycache__/epi_multitype.cpython-310.pyc +0 -0
  70. bella_companion/simulations/scenarios/__pycache__/epi_skyline.cpython-310.pyc +0 -0
  71. bella_companion/simulations/scenarios/__pycache__/fbd_2traits.cpython-310.pyc +0 -0
  72. bella_companion/simulations/scenarios/__pycache__/fbd_no_traits.cpython-310.pyc +0 -0
  73. bella_companion/simulations/scenarios/__pycache__/scenario.cpython-310.pyc +0 -0
  74. bella_companion/utils/__pycache__/__init__.cpython-310.pyc +0 -0
  75. bella_companion/utils/__pycache__/beast.cpython-310.pyc +0 -0
  76. bella_companion/utils/__pycache__/explain.cpython-310.pyc +0 -0
  77. bella_companion/utils/__pycache__/slurm.cpython-310.pyc +0 -0
  78. bella_companion/utils/explain.py +0 -45
  79. bella_companion-0.0.47.dist-info/RECORD +0 -116
  80. bella_companion-0.0.47.dist-info/WHEEL +0 -5
  81. bella_companion-0.0.47.dist-info/top_level.txt +0 -1
  82. /bella_companion/eucovid/data/{predictors/changetimes_flights_4e.tsv → changetimes.csv} +0 -0
  83. /bella_companion/eucovid/data/{predictors/flight_pop_x_4e_ls.tsv → flights.csv} +0 -0
@@ -0,0 +1,20 @@
1
+ from typing import Any, TypeGuard
2
+
3
+ import numpy as np
4
+ from bella_lumiere.typing import Weights
5
+
6
+
7
+ def is_list(obj: Any) -> TypeGuard[list[Any]]:
8
+ return isinstance(obj, list)
9
+
10
+
11
+ def is_weights(obj: Any) -> TypeGuard[Weights]:
12
+ return is_list(obj) and all(isinstance(layer, np.ndarray) for layer in obj)
13
+
14
+
15
+ def is_weights_list(weights: Any) -> TypeGuard[list[Weights]]:
16
+ return is_list(weights) and all(is_weights(w) for w in weights)
17
+
18
+
19
+ def is_nested_weights_list(weights: Any) -> TypeGuard[list[list[Weights]]]:
20
+ return is_list(weights) and all(is_weights_list(w) for w in weights)
@@ -1,12 +1,15 @@
1
- Metadata-Version: 2.4
1
+ Metadata-Version: 2.3
2
2
  Name: bella-companion
3
- Version: 0.0.47
3
+ Version: 0.1.1
4
4
  Summary: Bayesian Evolutionary Layered Learning Architectures (BELLA) companion
5
- Requires-Python: >=3.10
6
- Description-Content-Type: text/markdown
5
+ Author: gabriele-marino
6
+ Author-email: gabriele-marino <gabmarino.8601@gmail.com>
7
7
  Requires-Dist: arviz>=0.22.0
8
- Requires-Dist: bella-lumiere>=0.0.14
8
+ Requires-Dist: bella-lumiere>=0.1.0
9
9
  Requires-Dist: dotenv>=0.9.9
10
10
  Requires-Dist: phylogenie>=3.1.4
11
11
  Requires-Dist: poetry>=2.2.1
12
12
  Requires-Dist: seaborn>=0.13.2
13
+ Requires-Python: >=3.10
14
+ Description-Content-Type: text/markdown
15
+
@@ -0,0 +1,64 @@
1
+ bella_companion/.DS_Store,sha256=1lFlJ5EFymdzGAUAaI30vcaaLHt3F1LwpG7xILf9jsM,6148
2
+ bella_companion/BELLA.jar,sha256=4mKDzPfz263rTqA9KS2f4Mc-UWGMuj8G48ZDonKSqVw,67408597
3
+ bella_companion/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
+ bella_companion/cli.py,sha256=CrtV2CG6G1nKCCVu02gtNlpHNJHLnFtSaru-LFiHKKc,3014
5
+ bella_companion/eucovid/__init__.py,sha256=09Ld_G7fZvCDCgEWsmaOyLVQ8pFc9QHAGnXKJ9Zm2AM,81
6
+ bella_companion/eucovid/beast_configs/GLM.xml,sha256=-SZWxJgJdBthp132ckPpRGooKoYfKKREi6VV3pF22aY,11532
7
+ bella_companion/eucovid/beast_configs/MLP.xml,sha256=C2d-Mcjw44CLFL-pCYgYdbZb1bNR-6OUBtaLWLUt5AU,10539
8
+ bella_companion/eucovid/data/changetimes.csv,sha256=XEepIXKRwcKknJq4-ovKlorvCypGi67B8znuQ5f1LTg,60
9
+ bella_companion/eucovid/data/flights.csv,sha256=tMR9-tuGtC3rVABXUtNdV9irUX-9aCCVb_fwksHTn8U,1068
10
+ bella_companion/eucovid/data/msa.fasta,sha256=sWaVSINSJDDLjmwlpl31JWM5VnKF7yViSpERI69glQc,4839354
11
+ bella_companion/eucovid/run_beast.py,sha256=O9jULAMYIJOtQ_o4EQlG0Xh8s1Z2i27wmnGjYbvNWeM,1331
12
+ bella_companion/features.py,sha256=xj0iwGHhLV_4jxLmG0M-jXWejuKVFJqpB0dK8Skc9n8,283
13
+ bella_companion/platyrrhine/__init__.py,sha256=gySgNPoKaEdhcZppKMsCmS-aAcYw_rDEw0viz72Udaw,259
14
+ bella_companion/platyrrhine/beast_config.xml,sha256=bLzavt-Zd8cmSgQQnlCBfZ9KyM7d1inOGx12VNjPFb4,7056
15
+ bella_companion/platyrrhine/data/change_times.csv,sha256=tedobHbaY7-6M3V6F6HSqGAU9dpeeajR5mu1XGJfp6w,68
16
+ bella_companion/platyrrhine/data/traits.csv,sha256=N_mCnaM9Gy0Qot7-l_mK0vMvr7e_5BZ4bgUHpcoG9UM,3949
17
+ bella_companion/platyrrhine/data/trees.nwk,sha256=lpSiGUG5s3zE--IKPOyWrJ5vpJb_xAPnfxychs3YNIs,488232
18
+ bella_companion/platyrrhine/results.py,sha256=aRx9MkI43f5bhr-16UkShDH-_oMNFzhHxfU2WK1L3BY,7468
19
+ bella_companion/platyrrhine/run_beast.py,sha256=t1gUmn_HEXKqxgGXYCdohGDWeL-QEpW7r3bKXlRfdog,2494
20
+ bella_companion/platyrrhine/summarize.py,sha256=Qc3GE2vUspTqU_QjLdULvK0r7AZ3JTdqsBmd71izt7M,1356
21
+ bella_companion/simulations/__init__.py,sha256=gjO_QB2wp335l7UO-zA9fFQ5vmZauGItBuO-QZFu93E,439
22
+ bella_companion/simulations/beast_configs/epi-multitype/GLM.xml,sha256=VymAYsaZVuB3EPE0DCQSXpKepVq8pPjKaB5yhEpaVkE,4334
23
+ bella_companion/simulations/beast_configs/epi-multitype/MLP.xml,sha256=AoJyjvVcwKtut64d7sB2gnnenx5sXyuTTNCivykUyqU,4077
24
+ bella_companion/simulations/beast_configs/epi-multitype/Nonparametric.xml,sha256=DfCMcLM53b-XP9oIiQ9wJWOVTEP3GnoZaL5BMzLLfIQ,3208
25
+ bella_companion/simulations/beast_configs/epi-skyline/GLM.xml,sha256=7LtJhU7o7aQsnWK7ZIKj-mzmHgCn_RjxvKKbLIlp1Go,3619
26
+ bella_companion/simulations/beast_configs/epi-skyline/MLP.xml,sha256=ptoLM0FLDp54bUSxPm12qIvOYWyBkWR1e9VWVWjgU8E,3426
27
+ bella_companion/simulations/beast_configs/epi-skyline/Nonparametric.xml,sha256=LTiatqeAMiFfdD1leQ5TrGaJhsYwlkbMDpxqhVFOgUQ,2561
28
+ bella_companion/simulations/beast_configs/fbd-2traits/GLM.xml,sha256=Iff0eAShKLeOnlTmNu6044Wljtla3yckqo2LZ7mP6LY,5897
29
+ bella_companion/simulations/beast_configs/fbd-2traits/MLP.xml,sha256=RvF70fcnvoBcJxjyT4nELlFm94atSZpKm9d1aS-cJZ4,5428
30
+ bella_companion/simulations/beast_configs/fbd-2traits/Nonparametric.xml,sha256=ev6ZfMeCwMyK54G_4O8g-tbgoxoD3asxt71x91Cb1QY,3618
31
+ bella_companion/simulations/beast_configs/fbd-no-traits/GLM.xml,sha256=Caq7Gxqx-wYAu3_YkMikVqni3DjBv1Wt00sc1Upo1X8,4743
32
+ bella_companion/simulations/beast_configs/fbd-no-traits/MLP.xml,sha256=JE_cGh2DW5yPAXyXXr0D3DlovmV-aIbze7qrhENHK3w,4275
33
+ bella_companion/simulations/beast_configs/fbd-no-traits/Nonparametric.xml,sha256=m2xaHZwskCFInvVY1QH1fwQqH6p_fgESlPX2BbBeUhk,2822
34
+ bella_companion/simulations/figures/__init__.py,sha256=LeHPeBb4vxJo0pxLZi_9s6iRBKvWC32yhAZBByqK2_8,590
35
+ bella_companion/simulations/figures/epi_multitype.py,sha256=ez-yCP_rs7J0CkfEjt4SFwaiI5T0r3GcLhUXjxc9F3M,2627
36
+ bella_companion/simulations/figures/epi_skyline.py,sha256=SyFOsb8dwLuoN5IOCiQbG-ZCPDS8Wua_LL-cOuLtu_s,1673
37
+ bella_companion/simulations/figures/fbd_2traits.py,sha256=MCzZLmVi-txEIk8Sdm3Tw6oUGMahnifzMuPzwaDw_sY,2726
38
+ bella_companion/simulations/figures/fbd_no_traits.py,sha256=-_c-qnrFAYZ7vd4Ieeb-uYA3VUwURfrcP11BN9PG8JY,1907
39
+ bella_companion/simulations/figures/scenarios.py,sha256=gbMz1TUxxT2RSIq2kQlFioNdgSHk-gQY2OQuf6-7Fww,2817
40
+ bella_companion/simulations/figures/utils.py,sha256=0M5OrxaEuqcj9rR2uAc_O7utQvhEceZGH0sKrGRWaWs,3129
41
+ bella_companion/simulations/generate_data.py,sha256=edfhXrs76hl30i_xizup4z6qgzXZyaDzX_PVC62Ytpw,849
42
+ bella_companion/simulations/generate_figures.py,sha256=r3zGKnLGDpuVzzWVQIumkt1sTMl_4w1iKnVNy-fTJAg,371
43
+ bella_companion/simulations/metrics.py,sha256=kSX7-B4a-LDm1V--S8CvEJDktlJ8qkpo_gY6V7XbOXw,1930
44
+ bella_companion/simulations/run_beast.py,sha256=-L5PHkpPK-NGQoXxmeH7LMIWxtodvL4TOo-KkR2ISVQ,3299
45
+ bella_companion/simulations/scenarios/__init__.py,sha256=3Kl1lKcFpfb3vLX64DmSW4XCF5kXU1ZoHtstFH-ZIzU,876
46
+ bella_companion/simulations/scenarios/common.py,sha256=_ddaSuTvEVdttGkXB4HPc2B7IB1F_GBOCW3cVOPZ-ZM,807
47
+ bella_companion/simulations/scenarios/epi_multitype.py,sha256=hgl2-m9gPm6OntnOZeRfquflbXFuu2Kt4RmKinSslYo,2427
48
+ bella_companion/simulations/scenarios/epi_skyline.py,sha256=iLwsGw-DZK8QOIGhMRUyvFRDEwFrksSx46s2DB6sr0U,2353
49
+ bella_companion/simulations/scenarios/fbd_2traits.py,sha256=-BNJMx-4GGW32IaTDxwVOHs2ZW7z781GJn-tMycygcI,3529
50
+ bella_companion/simulations/scenarios/fbd_no_traits.py,sha256=OsJP2bJVswLgeerXSM0ULKnj5_lxRVHCCoENjs_fWPM,2508
51
+ bella_companion/simulations/scenarios/scenario.py,sha256=oV9yxxD_De5YdCLbAwIFLmiRXqOaoflJjTIkY2yu5M8,608
52
+ bella_companion/simulations/summarize.py,sha256=hfYpgmv2irD8wNQ-GYtJA324w94SKoYhx0bpDG2bb3w,1207
53
+ bella_companion/utils/__init__.py,sha256=Ehu8rgO6T2ie_BHW25I0-huoBIHPtknkHYvoUr-37BY,446
54
+ bella_companion/utils/beast.py,sha256=18LAAFkZeIvOGdA3VfTEjQxZKzDBW93OixGtMNZrRxU,2196
55
+ bella_companion/utils/explain/__init__.py,sha256=ksk8HWIpdiMhT4lW7H_05p_s44ihMiHHNM2TJGO9vz4,234
56
+ bella_companion/utils/explain/pdp.py,sha256=z7PhwnERDUnsM1XzYq971pjxadbgBhhaO7ZOwY0fKuM,5125
57
+ bella_companion/utils/explain/shap.py,sha256=ueCymQiVOcLIBy_Pf0yq7aOlI8ATehzhoxnBPYk8r0Y,3453
58
+ bella_companion/utils/explain/typeguards.py,sha256=U6N_lBosmXqFAiakB3zkjlHXzKd16e_cYkHLh76SSFY,591
59
+ bella_companion/utils/slurm.py,sha256=xEyf-Jxk8fy71t3fko_Ic9WtUFSdLFE3w4tR17gaBPw,1868
60
+ bella_companion/version.xml,sha256=Gr3nsYqFhVNP0Mszl_IfilxefsPN8ZpQlGlB8OJSwnE,522
61
+ bella_companion-0.1.1.dist-info/WHEEL,sha256=M6du7VZflc4UPsGphmOXHANdgk8zessdJG0DBUuoA-U,78
62
+ bella_companion-0.1.1.dist-info/entry_points.txt,sha256=Jmns5P9G0EXcV26VD5Na81SD2Fri_SiUISy9Rfz5ZYI,52
63
+ bella_companion-0.1.1.dist-info/METADATA,sha256=f_dbV3ST1DaBgVxI2t-cMnTN9U5197hPA8yGf4HN38U,472
64
+ bella_companion-0.1.1.dist-info/RECORD,,
@@ -0,0 +1,4 @@
1
+ Wheel-Version: 1.0
2
+ Generator: uv 0.9.5
3
+ Root-Is-Purelib: true
4
+ Tag: py3-none-any
@@ -1,2 +1,3 @@
1
1
  [console_scripts]
2
2
  bella = bella_companion.cli:main
3
+
@@ -1,399 +0,0 @@
1
- <?xml version="1.0" encoding="UTF-8" standalone="no"?>
2
- <beast namespace="beast.base.inference
3
- :beast.base.inference.parameter
4
- :beast.base.inference.distribution
5
- :beast.base.evolution.branchratemodel
6
- :beast.base.evolution.substitutionmodel
7
- :beast.base.evolution.sitemodel
8
- :beast.base.evolution.tree.coalescent
9
- :beast.base.evolution.tree
10
- :beast.base.evolution.likelihood
11
- :beast.base.evolution.alignment
12
- :beast.base.evolution.operator.kernel
13
- :beast.base.inference.operator.kernel
14
- :beast.base.inference.operator
15
- :beast.base.math.distributions
16
- :beast.base.evolution
17
- :beast.base.evolution.operator
18
- :bdmmprime.util
19
- :bdmmprime.util.operators
20
- :bdmmprime.util.priors
21
- :bdmmprime.distribution
22
- :bdmmprime.parameterization
23
- :bdmmprime.mapping
24
- :bdmmprime.trajectories
25
- :feast
26
- :feast.parameter
27
- :feast.fileio
28
- :feast.function
29
- :feast.expressions
30
- :glmprior.util"
31
- required="BEAST.base v2.7.7:BDMM-Prime v2.2.2" version="2.7">
32
-
33
-
34
- <!-- Map elements -->
35
- <map name="Uniform" >beast.base.inference.distribution.Uniform</map>
36
- <map name="Exponential" >beast.base.inference.distribution.Exponential</map>
37
- <map name="LogNormal" >beast.base.inference.distribution.LogNormalDistributionModel</map>
38
- <map name="prior" >beast.base.inference.distribution.Prior</map>
39
-
40
- <!-- Adapted from https://github.com/SarahNadeau/cov-europe-bdmm/blob/master/analyses/with_hubei_migration_decrease/with_hubei_migration_decrease_scrubbed.xml -->
41
- <!-- Sequence data loaded from GISAID with the gisaid EPI ISLs -->
42
-
43
- <!-- Alignment -->
44
- <alignment id="alignment" spec= "AlignmentFromFasta"
45
- fileName = "$(aligned_fasta)"/>
46
-
47
- <typeSet id="typeSetBDMMPrime" spec="TypeSet">
48
- <typeTraitSet id="typeTraitSet" spec="TraitSetFromTaxonSet"
49
- traitname="type"
50
- delimiter="|"
51
- takeGroup="1">
52
- <taxa spec="TaxonSet" alignment="@alignment"/>
53
- </typeTraitSet>
54
- </typeSet>
55
-
56
- <trait id="dateTrait" spec="TraitSetFromTaxonSet"
57
- traitname="date"
58
- dateFormat="yyyy-M-dd"
59
- delimiter= "|"
60
- everythingAfterLast="true">
61
- <taxa spec="TaxonSet" alignment="@alignment"/>
62
- </trait>
63
-
64
- <!-- BDMMPrime Population Model -->
65
-
66
- <processLength id="originBDMMPrime" spec="RealParameter">1.0</processLength>
67
-
68
- <Re id="ReSV" spec="SkylineVectorParameter"
69
- timesAreAges="true" processLength="@originBDMMPrime" typeSet="@typeSetBDMMPrime">
70
- <skylineValues id="ReValues" spec="RealParameter">1.0 1.1 1.2 1.3 1.4 1.01 1.1 1.2 1.3 1.4</skylineValues>
71
- <changeTimes id="ReRateChangeTimes" spec="RealParameter">0.123</changeTimes>
72
- </Re>
73
-
74
- <samplingProportion id="samplingProportionSV" spec="SkylineVectorParameter"
75
- timesAreAges="true" processLength="@originBDMMPrime" typeSet="@typeSetBDMMPrime">
76
- <skylineValues id="samplingProportionValues" spec="RealParameter" dimension="15" lower="0.0" upper="1.0">1.11E-5 1.2E-5 1.3E-5 1.4E-5 1.5E-5 1.1E-5 1.2E-5 1.3E-5 1.4E-5 1.5E-5 0.0 0.0 0.0 0.0 0.0</skylineValues>
77
-
78
-
79
- <changeTimes id="samplingProportionChangeTimes" spec="RealParameter">0.12 0.205</changeTimes>
80
- </samplingProportion>
81
-
82
-
83
- <becomeUninfectiousRate id="becomeUninfectiousRateSV" spec="SkylineVectorParameter"
84
- typeSet="@typeSetBDMMPrime">
85
- <skylineValues id="becomeUninfectiousValues" spec="RealParameter">36.5</skylineValues>
86
- </becomeUninfectiousRate>
87
-
88
-
89
- <migrationRate id="migrationRateSM" spec="SkylineMatrixParameter"
90
- timesAreAges="true" processLength="@originBDMMPrime" typeSet="@typeSetBDMMPrime">
91
- <skylineValues id="migrationRateValues_scaled" name="skylineValues" spec="feast.expressions.ExpCalculator">
92
- <![CDATA[
93
- migrationRateValues *
94
- { f,f,f,f,
95
- f, 1,1,1,
96
- f,1, 1,1,
97
- f,1,1, 1,
98
- f,1,1,1,
99
-
100
- 1,1,1,1,
101
- 1, 1,1,1,
102
- 1,1, 1,1,
103
- 1,1,1, 1,
104
- 1,1,1,1 }
105
- ]]>
106
- <arg id="migrationRateValues" spec="RealParameter" dimension="40">
107
- 0.11 0.15 0.2 0.24
108
- 0.0 0.14 0.18 0.22
109
- 0.0 0.1 0.19 0.23
110
- 0.0 0.12 0.16 0.25
111
- 0.0 0.13 0.17 0.21
112
-
113
- 0.11 0.15 0.2 0.24
114
- 0.0 0.14 0.18 0.22
115
- 0.0 0.1 0.19 0.23
116
- 0.0 0.12 0.16 0.25
117
- 0.0 0.13 0.17 0.21
118
- </arg>
119
- <arg id="f" spec="RealParameter">0.5</arg>
120
- </skylineValues>
121
-
122
- <changeTimes id="migrationRateChangeTimes" spec="RealParameter">0.123</changeTimes>
123
- </migrationRate>
124
-
125
-
126
- <removalProb id="removalProbSV" spec="SkylineVectorParameter" skylineValues="1.0"
127
- processLength="@originBDMMPrime" typeSet="@typeSetBDMMPrime"/>
128
-
129
- <startTypePriorProbs id="typeFrequencies" spec="RealParameter"
130
- estimate="false" lower="0.0" upper="1.0">1.0 0.0 0.0 0.0 0.0</startTypePriorProbs>
131
-
132
-
133
- <!-- Site Model -->
134
- <siteModel id="siteModel" spec="SiteModel" gammaCategoryCount="4">
135
- <parameter id="gammaShape" name="shape" value="1.0"/>
136
- <parameter name="mutationRate" estimate="false" value="1.0"/>
137
- <parameter name="proportionInvariant" estimate="false" value="0.0"/>
138
-
139
- <substModel spec="HKY">
140
- <kappa id="kappa" spec="RealParameter" value="2.0"/>
141
- <frequencies id="empiricalFreqs" spec="Frequencies" data="@alignment"/>
142
- </substModel>
143
- </siteModel>
144
-
145
-
146
- <!-- Branch rate Model -->
147
- <branchRateModel id="branchRateModel" spec="StrictClockModel">
148
- <parameter id="clockRate" name="clock.rate" estimate="false">8.0E-4</parameter>
149
- </branchRateModel>
150
-
151
-
152
- <!-- Population Model for Tree initialiser -->
153
- <popFunc id="popFunc" spec="ConstantPopulation">
154
- <popSize spec="RealParameter" value="0.1"/>
155
- </popFunc>
156
-
157
-
158
- <!-- MCMC -->
159
- <run id="mcmc" spec="MCMC" chainLength="10000000" numInitializationAttempts="100">
160
-
161
- <!-- State -->
162
- <state id="state" spec="State" storeEvery="5000">
163
-
164
- <stateNode id="treeBDMMPrime" spec="RandomTree"
165
- taxa="@alignment"
166
- populationModel="@popFunc"
167
- trait="@dateTrait">
168
- </stateNode>
169
-
170
- <stateNode idref="originBDMMPrime"/>
171
- <stateNode idref="ReValues"/>
172
- <stateNode idref="samplingProportionValues"/>
173
- <stateNode idref="migrationRateValues"/>
174
- <stateNode idref="f"/>
175
-
176
- <stateNode idref="kappa"/>
177
- <stateNode idref="gammaShape"/>
178
-
179
- </state>
180
-
181
-
182
- <!-- Posterior Distribution -->
183
- <distribution id="posterior" spec="CompoundDistribution">
184
-
185
- <!-- Tree Likelihood -->
186
- <distribution id="likelihood" spec="CompoundDistribution" useThreads="true">
187
-
188
- <distribution id="treeLikelihood" spec="ThreadedTreeLikelihood"
189
- data="@alignment"
190
- tree="@treeBDMMPrime"
191
- siteModel="@siteModel"
192
- branchRateModel="@branchRateModel"/>
193
-
194
- </distribution>
195
-
196
- <!-- Prior distribution -->
197
- <distribution id="prior" spec="CompoundDistribution">
198
-
199
- <!-- BDMM-Prime tree prior distribution -->
200
- <distribution id="BDMMPrime" spec="BirthDeathMigrationDistribution"
201
- tree="@treeBDMMPrime"
202
- conditionOnSurvival="true"
203
- finalSampleOffset="0.0"
204
- typeTraitSet="@typeTraitSet"
205
- startTypePriorProbs="@typeFrequencies">
206
-
207
- <!-- Parameterization BDMM-Prime -->
208
- <parameterization id="EpiBDMMPrimeParameterization" spec="EpiParameterization"
209
- processLength="@originBDMMPrime"
210
- Re="@ReSV"
211
- samplingProportion="@samplingProportionSV"
212
- becomeUninfectiousRate="@becomeUninfectiousRateSV"
213
- migrationRate="@migrationRateSM"
214
- removalProb="@removalProbSV"
215
- typeSet="@typeSetBDMMPrime"/>
216
-
217
- </distribution>
218
-
219
-
220
- <!-- Priors on parameters -->
221
- <!-- BDMMPrime parameters priors -->
222
-
223
- <prior id="originBDMMPrimePrior" name="distribution" x="@originBDMMPrime">
224
- <LogNormal name="distr" M="-1.0" S="0.2"/>
225
- </prior>
226
-
227
- <!-- Reproductive number Prior -->
228
- <distribution id="RePrior" spec="SmartZeroExcludingPrior" x="@ReValues">
229
- <LogNormal name="distr" M="0.8" S="0.5"/>
230
- </distribution>
231
-
232
- <!-- Prior for sampling proportion for China -->
233
- <distribution id="spChinaE2Prior" spec="SmartZeroExcludingPrior" x="@samplingProportionValues"
234
- classesToExclude="1.2E-5 1.3E-5 1.4E-5 1.5E-5">
235
- <Uniform name="distr" lower="0" upper="0.15"/>
236
- </distribution>
237
- <!-- Prior for sampling proportion for France -->
238
- <distribution id="spFranceE1Prior" spec="SmartZeroExcludingPrior" x="@samplingProportionValues"
239
- classesToExclude="1.1E-5 1.3E-5 1.4E-5 1.5E-5">
240
- <Uniform name="distr" lower="0" upper="0.093"/>
241
- </distribution>
242
- <!-- Prior for sampling proportion for Germany Epoch -->
243
- <distribution id="spGermanyE1Prior" spec="SmartZeroExcludingPrior" x="@samplingProportionValues"
244
- classesToExclude="1.1E-5 1.2E-5 1.4E-5 1.5E-5">
245
- <Uniform name="distr" lower="0" upper="0.10"/>
246
- </distribution>
247
- <!-- Prior for sampling proportion for Italy -->
248
- <distribution id="spItalyE1Prior" spec="SmartZeroExcludingPrior" x="@samplingProportionValues"
249
- classesToExclude="1.1E-5 1.2E-5 1.3E-5 1.5E-5">
250
- <Uniform name="distr" lower="0" upper="0.005"/>
251
- </distribution>
252
- <!-- Prior for sampling proportion for OtherEuropean -->
253
- <distribution id="spOEE1Prior" spec="SmartZeroExcludingPrior" x="@samplingProportionValues"
254
- classesToExclude="1.1E-5 1.2E-5 1.3E-5 1.4E-5">
255
- <Uniform name="distr" lower="0" upper="0.057"/>
256
- </distribution>
257
-
258
- <!-- Migration rate Prior -->
259
- <distribution id="migRatePrior" name="distribution" spec="SmartZeroExcludingPrior" x="@migrationRateValues">
260
- <LogNormal name="distr" M="0.0" S="1.0"/>
261
- </distribution>
262
-
263
- <distribution id="fPrior" name="distribution" spec="ZeroExcludingPrior" x="@f">
264
- <Uniform name="distr" lower="0.0" upper="1.0"/>
265
- </distribution>
266
-
267
- <!-- Substitution model parameter priors -->
268
- <prior id="GammaShapePrior" name="distribution" x="@gammaShape">
269
- <Exponential name="distr">
270
- <parameter spec="RealParameter" estimate="false" name="mean">0.5</parameter>
271
- </Exponential>
272
- </prior>
273
-
274
- <prior id="KappaPrior" name="distribution" x="@kappa">
275
- <LogNormal name="distr">
276
- <parameter spec="RealParameter" estimate="false" name="M">1.0</parameter>
277
- <parameter spec="RealParameter" estimate="false" name="S">1.25</parameter>
278
- </LogNormal>
279
- </prior>
280
-
281
- </distribution>
282
- </distribution>
283
-
284
-
285
- <!-- Operators -->
286
- <!-- Tree operators -->
287
- <operator id="BDMMPrimeTreeRootScaler" spec="BactrianScaleOperator" tree="@treeBDMMPrime"
288
- rootOnly="true" scaleFactor="0.25" weight="5.0"/>
289
-
290
- <operator id="BDMMPrimeUniformOperator" spec="BactrianNodeOperator" tree="@treeBDMMPrime"
291
- weight="30.0"/>
292
-
293
- <operator id="BDMMPrimeSubtreeSlideScaler" spec="BactrianSubtreeSlide" tree="@treeBDMMPrime"
294
- weight="15.0"/>
295
-
296
- <operator id="BDMMPrimeNarrow" spec="Exchange" tree="@treeBDMMPrime"
297
- weight="15.0"/>
298
-
299
- <operator id="BDMMPrimeWide" spec="Exchange" tree="@treeBDMMPrime"
300
- isNarrow="false" weight="3.0"/>
301
-
302
- <operator id="BDMMPrimeWilsonBalding" spec="WilsonBalding" tree="@treeBDMMPrime"
303
- weight="3.0"/>
304
-
305
- <operator id="BDMMPrimeBICEPSEpochTop" spec="EpochFlexOperator" tree="@treeBDMMPrime"
306
- scaleFactor="0.1" weight="2.0"/>
307
-
308
- <operator id="BDMMPrimeBICEPSEpochAll" spec="EpochFlexOperator" tree="@treeBDMMPrime"
309
- fromOldestTipOnly="false" scaleFactor="0.1" weight="2.0"/>
310
-
311
- <operator id="BDMMPrimeBICEPSTreeFlex" spec="TreeStretchOperator" tree="@treeBDMMPrime"
312
- scaleFactor="0.01" weight="2.0"/>
313
-
314
-
315
- <!-- Population model operators -->
316
- <operator id="originBDMMPrimeScaler" spec="BactrianScaleOperator" parameter="@originBDMMPrime"
317
- scaleFactor="0.25" weight="5.0" />
318
-
319
- <!-- <operator id="ReScaler" spec="BactrianScaleOperator" parameter="@ReValues"
320
- weight="5.0" />
321
- <operator id="ReScalerAll" spec="BactrianScaleOperator" parameter="@ReValues"
322
- weight="5.0" scaleAll="true" /> -->
323
-
324
- <operator id="ReScaler" spec="SmartScaleOperator" parameter="@ReValues"
325
- weight="5.0" />
326
-
327
- <operator id="samplingProportionScaler" spec="SmartScaleOperator" parameter="@samplingProportionValues"
328
- weight="3.0" />
329
-
330
- <operator id="migrationRateScaler" spec="SmartScaleOperator" parameter="@migrationRateValues"
331
- weight="5.0" />
332
- <operator id="fScaler" spec="BactrianScaleOperator" parameter="@f"
333
- weight="5.0"/>
334
-
335
- <!-- Substitution model operators -->
336
- <operator id="gammaShapeScaler" spec="BactrianScaleOperator" parameter="@gammaShape"
337
- weight="1.0"/>
338
-
339
- <operator id="KappaScaler" spec="BactrianScaleOperator" parameter="@kappa"
340
- weight="1.0"/>
341
-
342
-
343
- <!-- Loggers -->
344
- <logger id="tracelog" spec="Logger" fileName="$(filebase).log" logEvery="1000" model="@posterior" sanitiseHeaders="true" sort="smart">
345
- <log idref="posterior"/>
346
- <log idref="likelihood"/>
347
- <log idref="prior"/>
348
- <log idref="treeLikelihood"/>
349
- <log idref="gammaShape"/>
350
- <log idref="kappa"/>
351
- <log id="TreeHeight" spec="TreeStatLogger" tree="@treeBDMMPrime"/>
352
- <log id="typedTreeStats" spec="TypedTreeStatsLogger" includeRootEdge="true"
353
- typeLabel="type" typeSet="@typeSetBDMMPrime">
354
- <typedTree id="typeMappedTree" spec="TypeMappedTree" bdmmDistrib="@BDMMPrime" startTypePriorProbs="@typeFrequencies" mapOnInit="false" remapOnLog="true" typeLabel="type" typeTraitSet="@typeTraitSet" untypedTree="@treeBDMMPrime"/>
355
- </log>
356
- <log idref="originBDMMPrime"/>
357
- <log idref="ReSV"/>
358
- <log idref="becomeUninfectiousRateSV"/>
359
- <log idref="samplingProportionSV"/>
360
- <log idref="migrationRateSM"/>
361
- <log idref="migrationRateValues"/>
362
- <log idref="f"/>
363
- <log idref="removalProbSV"/>
364
- <log idref="typeFrequencies"/>
365
- </logger>
366
-
367
- <logger id="screenlog" spec="Logger" logEvery="1000">
368
- <log idref="posterior"/>
369
- <log arg="@posterior" id="ESS_posterior" spec="util.ESS"/>
370
- <log idref="likelihood"/>
371
- <log arg="@likelihood" id="ESS_likelihood" spec="util.ESS"/>
372
- <log idref="prior"/>
373
- <log arg="@prior" id="ESS_prior" spec="util.ESS"/>
374
- <log idref="BDMMPrime"/>
375
- <log arg="@BDMMPrime" id="ESS_BDMMPrime" spec="util.ESS"/>
376
- </logger>
377
-
378
- <logger id="treelog" spec="Logger" fileName="$(filebase).trees" logEvery="1000" mode="tree">
379
- <log id="TreeWithMetaDataLogger" spec="TreeWithMetaDataLogger" tree="@treeBDMMPrime"/>
380
- </logger>
381
-
382
- <logger id="typedTreeLogger" spec="OptionalLogger" enableLogger="true"
383
- fileName="$(filebase).typed.trees" logEvery="10000" mode="tree">
384
- <log idref="typeMappedTree"/>
385
- </logger>
386
- <logger id="nodeTypedTreeLogger.t:eubdmm_alignment" spec="OptionalLogger" enableLogger="true"
387
- fileName="$(filebase).typed.node.trees" logEvery="10000" mode="tree">
388
- <log id="nodeTypedTree" spec="TypedNodeTreeLogger" typedTree="@typeMappedTree"/>
389
- </logger>
390
- <logger id="trajLogger" spec="OptionalLogger" enableLogger="true" fileName="$(filebase).traj" logEvery="10000">
391
- <log id="typedTraj" spec="SampledTrajectory" bdmmDistrib="@BDMMPrime"
392
- startTypePriorProbs="@typeFrequencies" typeMappedTree="@typeMappedTree" useTauLeaping="true"/>
393
- </logger>
394
-
395
- <operatorschedule id="OperatorSchedule" spec="OperatorSchedule"/>
396
-
397
- </run>
398
-
399
- </beast>