diff options
Diffstat (limited to 'simulation/mle_blocks.py')
| -rw-r--r-- | simulation/mle_blocks.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/simulation/mle_blocks.py b/simulation/mle_blocks.py index 0d27869..98bc257 100644 --- a/simulation/mle_blocks.py +++ b/simulation/mle_blocks.py @@ -54,8 +54,10 @@ if __name__ == "__main__": bm.TrainingDataMonitoring([cost, params, 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), + ub.JSONDump("logs/active_estoutdegree_mle.json", + every_n_batches=freq), + ub.ActiveLearning(data_stream.dataset, params, + every_n_batches=freq), ], ) loop.run() |