Coverage for src/scicom/utilities/statistics.py: 0%

10 statements  

« prev     ^ index     » next       coverage.py v7.4.4, created at 2024-04-15 13:26 +0200

1import os 

2from tqdm import tqdm 

3import pandas as pd 

4 

5from semanticlayertools.metric.prune import PruneNetwork 

6 

7def prune( 

8 modelparameters: dict, 

9 network: tuple, 

10 columns: list, 

11 iterations=10, 

12 delAmounts=(0.1, 0.25, 0.5, 0.75, 0.9), 

13 delTypes=("degree", "unif") 

14): 

15 """Generate pruned networks from input. 

16  

17 Assumes existence of columns "sender", "receiver", and "step". 

18 """ 

19 runDf = pd.DataFrame(network, columns = columns) 

20 pruning = PruneNetwork(runDf) 

21 result = pruning.deleteFromNetwork( 

22 iterations=iterations, 

23 delAmounts=delAmounts, 

24 delTypes=delTypes 

25 ) 

26 result = result.assign(**modelparameters) 

27 return result 

28