multiml.agent.keras.keras_darts module

class multiml.agent.keras.keras_darts.KerasDartsAgent(select_one_models=True, use_original_darts_optimization=True, dartstask_args={}, **kwargs)

Bases: KerasEnsembleAgent

Model selection agent inspired by Differential Architecture Search (DARTS)

__init__(select_one_models=True, use_original_darts_optimization=True, dartstask_args={}, **kwargs)

Parameters:

• select_one_models (bool) – (Expert option) If false, all models are kept for the final prediction

• use_original_darts_optimization (bool) – (Expert option) If false, both model parameter and alpha parameters are simultaneously trained.

• dartstask_args (dict) – args for DARTSTask

• **kwargs – Arbitrary keyword arguments