diff --git a/data/parameters.yaml b/data/parameters.yaml index bea6872..faad2af 100644 --- a/data/parameters.yaml +++ b/data/parameters.yaml @@ -5,8 +5,8 @@ model_parameters: # Transmission dynamics - beta: 0.05 # Transmission rate per contact per day - sigma: 0.0002 # Progression rate from latent to active TB (1/day) + beta: 0.125 # Transmission rate per contact per day + sigma: 0.0001 # Progression rate from latent to active TB (1/day) gamma: 0.005 # Natural recovery rate from active TB (1/day) mu: 0.00004 # Natural death rate (1/day) diff --git a/data/parameters3.yaml b/data/parameters3.yaml new file mode 100644 index 0000000..bc8da00 --- /dev/null +++ b/data/parameters3.yaml @@ -0,0 +1,18 @@ + +model_parameters: + # Transmission dynamics + beta: 0.35 # Transmission rate per contact per day + sigma: 0.0002 # Progression rate from latent to active TB (1/day) + gamma: 0.005 # Natural recovery rate from active TB (1/day) + mu: 0.00004 # Natural death rate (1/day) + + # Intervention parameters + treatment_rate: 0.01 # Treatment initiation rate (1/day) + vaccine_efficacy: 0.7 # Vaccine effectiveness (proportion) + + # Contact and mixing parameters + contact_rate: 10 # Average contacts per person per day + + # Disease progression probabilities + prob_fast_progression: 0.05 # Probability of fast progression to active TB + prob_reactivation: 0.001 # Annual reactivation rate for latent TB diff --git a/results/tb_model_output.png b/results/tb_model_output.png index 428ad24..05120a6 100644 Binary files a/results/tb_model_output.png and b/results/tb_model_output.png differ diff --git a/src/tb_model.py b/src/tb_model.py index 86d16e2..56071f4 100644 --- a/src/tb_model.py +++ b/src/tb_model.py @@ -249,8 +249,10 @@ def run_simulation(self, years=10): print(f"Simulation completed for {years} years") return results - def plot_results(self, save_path='../results/tb_model_output.png'): - """Create visualization of model results.""" + def plot_results(self, save_path='./results/tb_model_output.png'): + """Create visualization of model results. + save.path""" + if self.results is None: print("No results to plot. Run simulation first.") return