mirror of
https://github.com/vale981/HOPSFlow-Paper
synced 2025-03-05 17:51:39 -05:00
fix integration helper
This commit is contained in:
parent
3f7ffdd9e1
commit
e6a0eb62e6
1 changed files with 14 additions and 2 deletions
|
@ -190,12 +190,24 @@ def integrate_online(model, n, stream_folder=None, **kwargs):
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def get_sample_count(model):
|
||||||
|
try:
|
||||||
|
with aux.get_data(model) as d:
|
||||||
|
return d.samples
|
||||||
|
|
||||||
|
except:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
def integrate_online_multi(models, n, *args, increment=1000, **kwargs):
|
def integrate_online_multi(models, n, *args, increment=1000, **kwargs):
|
||||||
target = increment
|
target = increment
|
||||||
|
|
||||||
while target < (n + target):
|
while target <= n:
|
||||||
|
current_target = min([n, target])
|
||||||
for model in models:
|
for model in models:
|
||||||
integrate_online(model, min([n, target]), *args, **kwargs)
|
count = get_sample_count(model)
|
||||||
|
if count < current_target:
|
||||||
|
integrate_online(model, current_target, *args, **kwargs)
|
||||||
|
|
||||||
target += increment
|
target += increment
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue