Coverage for kcpdi/utils.py: 83%

1import matplotlib.pyplot as plt 

2 

3 

4def fig_ax(figsize=(15, 5), dpi=150): 

5 """Generate a (matplotlib) figure and ax objects with given size.""" 

6 return plt.subplots(figsize=figsize, dpi=dpi) 

7 

8 

9def get_sum_of_cost(algo, n_bkps) -> float: 

10 """Calculate the sum of costs for the change points `bkps`. 

11 

12 Utility function used to do penalized variable selection and obtain 

13 a final list of anomaly time points. 

14 """ 

15 bkps = algo.predict(n_bkps=n_bkps) 

16 return algo.cost.sum_of_costs(bkps)