diff options
Diffstat (limited to 'simulation/mle_blocks.py')
| -rw-r--r-- | simulation/mle_blocks.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/simulation/mle_blocks.py b/simulation/mle_blocks.py index ab8816f..0d27869 100644 --- a/simulation/mle_blocks.py +++ b/simulation/mle_blocks.py @@ -30,8 +30,9 @@ def create_mle_model(graph): if __name__ == "__main__": batch_size = 100 - n_obs = 100000 - graph = utils.create_wheel(100) + #n_obs = 100000 + freq = 10 + graph = utils.create_wheel(1000) print('GRAPH:\n', graph, '\n-------------\n') @@ -51,10 +52,10 @@ if __name__ == "__main__": extensions=[ be.FinishAfter(after_n_batches=10**3), bm.TrainingDataMonitoring([cost, params, - rmse, error], every_n_batches=10), - be.Printing(every_n_batches=10), - ub.JSONDump("log.json", every_n_batches=10), - ub.ActiveLearning(data_stream.dataset), + rmse, error], every_n_batches=freq), + be.Printing(every_n_batches=freq), + ub.JSONDump("logs/active_outdegree_mle.json", every_n_batches=freq), + ub.ActiveLearning(data_stream.dataset, graph, every_n_batches=freq), ], ) loop.run() |