def test_track_the_best(): main_loop = MockMainLoop() extension = TrackTheBest("cost") extension.main_loop = main_loop main_loop.status['iterations_done'] += 1 main_loop.log.current_row['cost'] = 5 extension.dispatch('after_batch') assert main_loop.status['best_cost'] == 5 assert main_loop.log.current_row['cost_best_so_far'] main_loop.status['iterations_done'] += 1 main_loop.log.current_row['cost'] = 6 extension.dispatch('after_batch') assert main_loop.status['best_cost'] == 5 assert main_loop.log.current_row.get('cost_best_so_far', None) is None main_loop.status['iterations_done'] += 1 main_loop.log.current_row['cost'] = 5 extension.dispatch('after_batch') assert main_loop.status['best_cost'] == 5 assert main_loop.log.current_row.get('cost_best_so_far', None) is None main_loop.status['iterations_done'] += 1 main_loop.log.current_row['cost'] = 4 extension.dispatch('after_batch') assert main_loop.status['best_cost'] == 4 assert main_loop.log.current_row['cost_best_so_far']
def test_track_the_best(): main_loop = MockMainLoop() extension = TrackTheBest("cost") extension.main_loop = main_loop main_loop.status['epochs_done'] += 1 main_loop.status['iterations_done'] += 10 main_loop.log.current_row['cost'] = 5 extension.dispatch('after_epoch') assert main_loop.status['best_cost'] == 5 assert main_loop.log.current_row['cost_best_so_far'] main_loop.status['epochs_done'] += 1 main_loop.status['iterations_done'] += 10 main_loop.log.current_row['cost'] = 6 extension.dispatch('after_epoch') assert main_loop.status['best_cost'] == 5 assert main_loop.log.current_row.get('cost_best_so_far', None) is None main_loop.status['epochs_done'] += 1 main_loop.status['iterations_done'] += 10 main_loop.log.current_row['cost'] = 5 extension.dispatch('after_epoch') assert main_loop.status['best_cost'] == 5 assert main_loop.log.current_row.get('cost_best_so_far', None) is None main_loop.status['epochs_done'] += 1 main_loop.status['iterations_done'] += 10 main_loop.log.current_row['cost'] = 4 extension.dispatch('after_epoch') assert main_loop.status['best_cost'] == 4 assert main_loop.log.current_row['cost_best_so_far']
def test_track_the_best(): main_loop = MockMainLoop() extension = TrackTheBest("cost") extension.main_loop = main_loop main_loop.status["epochs_done"] += 1 main_loop.status["iterations_done"] += 10 main_loop.log.current_row["cost"] = 5 extension.dispatch("after_epoch") assert main_loop.status["best_cost"] == 5 assert main_loop.log.current_row["cost_best_so_far"] main_loop.status["epochs_done"] += 1 main_loop.status["iterations_done"] += 10 main_loop.log.current_row["cost"] = 6 extension.dispatch("after_epoch") assert main_loop.status["best_cost"] == 5 assert main_loop.log.current_row.get("cost_best_so_far", None) is None main_loop.status["epochs_done"] += 1 main_loop.status["iterations_done"] += 10 main_loop.log.current_row["cost"] = 5 extension.dispatch("after_epoch") assert main_loop.status["best_cost"] == 5 assert main_loop.log.current_row.get("cost_best_so_far", None) is None main_loop.status["epochs_done"] += 1 main_loop.status["iterations_done"] += 10 main_loop.log.current_row["cost"] = 4 extension.dispatch("after_epoch") assert main_loop.status["best_cost"] == 4 assert main_loop.log.current_row["cost_best_so_far"]
def test_track_the_best(): main_loop = MockMainLoop() extension = TrackTheBest("cost") extension.main_loop = main_loop main_loop.status.iterations_done += 1 main_loop.log.current_row.cost = 5 extension.dispatch('after_batch') assert main_loop.status.best_cost == 5 assert main_loop.log.current_row['cost_best_so_far'] main_loop.status.iterations_done += 1 main_loop.log.current_row.cost = 6 extension.dispatch('after_batch') assert main_loop.status.best_cost == 5 assert main_loop.log.current_row['cost_best_so_far'] is None main_loop.status.iterations_done += 1 main_loop.log.current_row.cost = 5 extension.dispatch('after_batch') assert main_loop.status.best_cost == 5 assert main_loop.log.current_row['cost_best_so_far'] is None main_loop.status.iterations_done += 1 main_loop.log.current_row.cost = 4 extension.dispatch('after_batch') assert main_loop.status.best_cost == 4 assert main_loop.log.current_row['cost_best_so_far']