def draw_survival_curves_and_histograms(calibrated_model_no_drug,
calibrated_model_with_drug):
""" draws the histograms of average survival time
:param calibrated_model_no_drug: calibrated model simulated when drug is not available
:param calibrated_model_with_drug: calibrated model simulated when drug is available
"""
# get survival curves of both treatments
survival_curves = [
calibrated_model_no_drug.multiCohorts.multiCohortOutcomes.
survivalCurves, calibrated_model_with_drug.multiCohorts.
multiCohortOutcomes.survivalCurves
]
# graph survival curve
Path.plot_sets_of_sample_paths(sets_of_sample_paths=survival_curves,
title='Survival curve',
x_label='Simulation time step',
y_label='Number of alive patients',
legends=['No Drug', 'With Drug'],
color_codes=['blue', 'orange'],
transparency=0.25)
# histograms of average survival times
set_of_survival_times = [
calibrated_model_no_drug.multiCohorts.multiCohortOutcomes.
meanSurvivalTimes, calibrated_model_with_drug.multiCohorts.
multiCohortOutcomes.meanSurvivalTimes
]
# graph histograms
Hist.plot_histograms(data_sets=set_of_survival_times,
title='Histogram of average patient survival time',
x_label='Survival time',
y_label='Counts',
bin_width=0.5,
legends=['No Drug', 'With Drug'],
color_codes=['blue', 'orange'],
transparency=0.5,
x_range=[6, 20])
def plot_survival_curves_and_histograms(multi_cohort_outcomes_mono,
multi_cohort_outcomes_combo):
""" plot the survival curves and the histograms of survival times
:param multi_cohort_outcomes_mono: outcomes of a multi-cohort simulated under mono therapy
:param multi_cohort_outcomes_combo: outcomes of a multi-cohort simulated under combination therapy
"""
# get survival curves of both treatments
sets_of_survival_curves = [
multi_cohort_outcomes_mono.survivalCurves,
multi_cohort_outcomes_combo.survivalCurves
]
# graph survival curve
Path.plot_sets_of_sample_paths(
sets_of_sample_paths=sets_of_survival_curves,
title='Survival Curves',
x_label='Simulation Time Step (year)',
y_label='Number of Patients Alive',
legends=['Mono Therapy', 'Combination Therapy'],
transparency=0.4,
color_codes=['green', 'blue'])
# histograms of survival times
set_of_survival_times = [
multi_cohort_outcomes_mono.meanSurvivalTimes,
multi_cohort_outcomes_combo.meanSurvivalTimes
]
# graph histograms
Hist.plot_histograms(data_sets=set_of_survival_times,
title='Histograms of Average Patient Survival Time',
x_label='Survival Time (year)',
y_label='Counts',
bin_width=0.25,
x_range=[5.25, 17.75],
legends=['Mono Therapy', 'Combination Therapy'],
color_codes=['green', 'blue'],
transparency=0.5)
title='Plotting a single sample path that is updated in batch',
x_label='time',
y_label='observed value',
legend='Path 3',
color_code='r')
# plot both paths
Path.plot_sample_paths(sample_paths=[path1, path2],
title='Plot two sample paths with different color',
x_label='time',
y_label='observed value',
legends=['Path 1', 'Path 2'],
transparency=0.75)
Path.plot_sample_paths(sample_paths=[path1, path2],
title='Plot 2 sample paths with the same color',
x_label='time',
y_label='observed value',
legends='Path',
transparency=0.5,
common_color_code='g')
Path.plot_sets_of_sample_paths(sets_of_sample_paths=[[path1, path2],
[path3, path4]],
title='Plot 2 sets of sample paths',
x_label='time',
y_label='observed value',
legends=['Set 1', 'Set 2'],
transparency=0.5,
color_codes=['g', 'b'])