starsimhub / stisim

Analyses of syphilis diagnostics
MIT License
0 stars 1 forks source link

Linear Sum Agreement Algorithm #1

Closed daniel-klein closed 4 months ago

daniel-klein commented 6 months ago

Experimenting with pair formation via linear sum agreement (LSA), as in EmbeddingNet.

Resulting pars are a bit different, LSA is on the right.

image
daniel-klein commented 6 months ago

Note, LSA will create at most one partner per person per step, unlike the previous algorithm, so would be good to check concurrency.

daniel-klein commented 6 months ago
image
daniel-klein commented 6 months ago

Main takes 10.37 s and this branch is 19.59 s, so about 2x slower with LSA.

daniel-klein commented 6 months ago

In the lifetime partners plot above, I'm surprise to see LSA making so many very high results. Not sure what's going on there!

robynstuart commented 4 months ago

Closing this for now, can re-open once repo is up to date with latest starsim