autointent.nodes.NodeOptimizer#

class autointent.nodes.NodeOptimizer(node_type, search_space, metric)#

Node optimizer class.

Parameters:
node_type#
node_info#
metric_name#
modules_search_spaces#
fit(context)#

Fit the node optimizer.

Parameters:

context (autointent.context.Context) – Context

Return type:

None

get_module_dump_dir(dump_dir, module_name, j_combination)#

Get module dump directory.

Parameters:

  • dump_dir (pathlib.Path) – The base directory where the module dump directories will be created.

  • module_name (str) – The type of the module being optimized.

  • j_combination (int) – The index of the parameter combination being used.

Returns:

The path to the module dump directory as a string.

Return type:

str

module_fit(module, context)#

Fit the module.

Parameters:
Return type:

None