ml4co_kit.learning.env

Base class for environments.

Classes

BaseEnv(name[, mode, train_path, val_path, ...])

class ml4co_kit.learning.env.BaseEnv(name: str, mode: str = 'train', train_path: str | None = None, val_path: str | None = None, test_path: str | None = None, train_batch_size: int = 1, val_batch_size: int = 1, test_batch_size: int = 1, num_workers: int = 0, device: str = 'cpu')

Bases: object

load_data()

test_dataloader()

train_dataloader()

val_dataloader()