multiml.agent.basic.connection_random_search module

class multiml.agent.basic.connection_random_search.ConnectionRandomSearchAgent(freeze_model_weights=False, do_pretraining=True, connectiontask_name=None, connectiontask_args={}, **kwargs)

Bases: RandomSearchAgent

Agent executing with only one possible hyperparameter combination.

__init__(freeze_model_weights=False, do_pretraining=True, connectiontask_name=None, connectiontask_args={}, **kwargs)

Parameters:

• freeze_model_weights (bool) – Fix model trainable weights after pre-training

• do_pretraining (bool) – run pre-training before model connection

• connectiontask_name (str) – Task name for ModelConnectionTask

• connectiontask_args (dict) – args for ModelConnectionTask

• **kwargs – Arbitrary keyword arguments

execute()

Execute.

connectiontask_name_with_jobid(job_id)

Returns a formatted name for ConnectionTask. job_id is used as a suffix.

Returns:

formatted name for the ConnectionTask

Return type:

str