1. 24 Jun, 2020 1 commit
    • Chris Jewell's avatar
      Changes: · e72cfe76
      Chris Jewell authored
      1. Formatting -- black style!
      2. After profiling, implemented a better workaround for the
      tfd.Categorical bug.  Replaced with an inherited Categorical2 class,
      with overloaded log_prob function.
      3. Fixed a  bug in EventTimeProposal where tfd.FiniteDiscrete was
      being used on a non-strictly-increasing outcomes vector.
      4. Increased number of meta-populations to 149 to mirror UK UTLA regions.
      e72cfe76
  2. 22 Jun, 2020 1 commit
    • Chris Jewell's avatar
      Changes: · 705ce40b
      Chris Jewell authored
      1. Fixed bug in covid.impl.event_time_mh._reverse_move where Multinomial one_hot tensor wasn't being advanced for the reverse move.  This is a temporary workaround while tf.nn.sparse_softmax_cross_entropy_with_logits is fixed.
      
      2. Batched EventTimesProposal to move multiple meta-populations.
      
      3. Batched _move_events to allow for multiple meta-population updates
      
      4. Refactored covid.impl.event_time_proposal._abscumdiff so we calculate `bound_times` outside the function.  This allows clipping of times proposed outside [0, T] to [0, T], thus avoiding an error in tf.gather_nd in CPU mode.  The output of EventTimesProposal.sample and EventTimesProposal.log_prob might be inconsistent with what we expect, but that's okay as the move gets rejected as an out-of-bounds proposal anyway.
      705ce40b
  3. 20 Jun, 2020 1 commit
  4. 19 Jun, 2020 3 commits
  5. 18 Jun, 2020 1 commit
    • Chris Jewell's avatar
      Changes: · ac099f62
      Chris Jewell authored
      1. Edited _move_events to take into account metapopulation selection.
      2. Replace bound_t.ndim with bound_t.shape.rank for TF Graph Mode running.
      3. Revised index calculation in `_abscumdiff`.  Re-wrote test to accommodate.
      4. Code rationalisation in FilteredEventTimesProposal to return a simpler data structure.
      5. Changed dimension of results.extra field to trace `x_star`.
      6. Edited tests in `tests.test_mcmc.py` for new `_abscumdiff` interface.
      ac099f62
  6. 16 Jun, 2020 3 commits
  7. 13 Jun, 2020 1 commit
  8. 12 Jun, 2020 3 commits
  9. 05 Jun, 2020 1 commit
  10. 04 Jun, 2020 1 commit
  11. 03 Jun, 2020 2 commits
  12. 01 Jun, 2020 1 commit
  13. 29 May, 2020 2 commits
  14. 24 May, 2020 1 commit
    • Chris Jewell's avatar
      Changes: · a0488186
      Chris Jewell authored
      1. Corrected bug in UncalibratedEventTimesUpdate.one_step, and factored out some propagation functions.
      2. Inserted a tf.cond into event times update step to deal with the case where there is no preceding or next event (e.g. S->E in an SEIR model doesn't have a preceding event).  This has reduced performance, and could possibly be replaced with a NOOP calculation instead.
      3. Removed call to MH_within_Gibbs -- this proved too restrictive when requiring more than one update to a state_part per sweep of the MHwG sampler.
      a0488186
  15. 21 May, 2020 2 commits
  16. 20 May, 2020 1 commit
  17. 19 May, 2020 2 commits
  18. 18 May, 2020 3 commits
  19. 12 May, 2020 2 commits
  20. 11 May, 2020 2 commits
  21. 06 May, 2020 1 commit
  22. 02 May, 2020 1 commit
    • Chris Jewell's avatar
      Major changes: · 1b317d5c
      Chris Jewell authored
      1. Removed TransformedTransitionKernel.  This kernel transforms the log_prob function, so it it no longer compatible with the general target_log_prob.
      
      2. Replaced with an UncalibratedLogRandomWalk class which is composed inside a MetropolisHastings kernel.
      1b317d5c
  23. 01 May, 2020 1 commit
    • Chris Jewell's avatar
      Changes: · dd1635d3
      Chris Jewell authored
      1. Implemented Haario-style covariance update for parameters
      2. Fixed a bug in the use of underlying tfp.mcmc framework where TransformedTransitionKernel.bootstrap_results was returning incorrect results.
      dd1635d3
  24. 28 Apr, 2020 1 commit
  25. 27 Apr, 2020 1 commit
  26. 23 Apr, 2020 1 commit