ml4co_kit.wrapper.base

Base class for all wrappers in the ML4CO kit.

Classes

WrapperBase(task_type, precision)

class ml4co_kit.wrapper.base.WrapperBase(task_type: ~ml4co_kit.task.base.TASK_TYPE, precision: ~numpy.float32 | ~numpy.float64 = <class 'numpy.float32'>)

Bases: object

evaluate(check_constr: bool = True) → float

Evaluate the task list.

evaluate_w_gap(check_constr: bool = True) → Sequence[float]

Evaluate the task list.

from_pickle(file_path: Path)

from_task_pickle_folder(task_class: Type[TaskBase], folder_path: Path)

from_txt(file_path: Path, *args, **kwargs)

generate(generator: GeneratorBase, solver: SolverBase, num_samples: int = 1280, num_threads: int = 1, batch_size: int = 1, optimizer_parallel: bool = False, show_time: bool = True)

generate_w_to_txt(file_path: Path, generator: GeneratorBase, solver: SolverBase, num_samples: int = 1280, num_threads: int = 1, batch_size: int = 1, write_per_iters: int = 1, write_mode: str = 'a', show_time: bool = True)

solve(solver: SolverBase, num_threads: int = 1, batch_size: int = 1, optimizer_parallel: bool = False, show_time: bool = False)

swap_sol_and_ref_sol()

to_pickle(file_path: Path)

to_task_pickle_folder(folder_path: Path)

to_txt(show_time: bool = False, mode: str = 'w')