def __init__(self,
final_layer,
dataset,
splitter,
preprocessor,
iterator,
loss_expression,
updates_expression,
updates_modifier,
monitors,
stop_criterion,
remember_best_chan,
run_after_early_stop,
batch_modifier=None):
self.final_layer = final_layer
self.dataset = dataset
self.dataset_provider = PreprocessedSplitter(splitter, preprocessor)
self.preprocessor = preprocessor
self.iterator = iterator
self.loss_expression = loss_expression
self.updates_expression = updates_expression
self.updates_modifier = updates_modifier
self.monitors = monitors
self.stop_criterion = stop_criterion
self.monitor_manager = MonitorManager(monitors)
self.remember_extension = RememberBest(remember_best_chan)
self.run_after_early_stop = run_after_early_stop
self.batch_modifier = batch_modifier
def setup(self, target_var=None):
lasagne.random.set_rng(RandomState(9859295))
self.dataset.ensure_is_loaded()
self.print_layer_sizes()
log.info("Create theano functions...")
self.create_theano_functions(target_var)
# reset remember best extension in case you rerun some experiment
self.remember_extension = RememberBest(
self.remember_extension.chan_name)
log.info("Done.")
