Function train_code2class

code2seq/code2class_wrapper.py:22–34

(config: DictConfig)

def train_code2class(config: DictConfig):
    filter_warnings()

    if config.print_config:
        print_config(config, fields=["model", "data", "train", "optimizer"])

    # Load data module
    data_module = PathContextDataModule(config.data_folder, config.data, is_class=True)

    # Load model
    code2class = Code2Class(config.model, config.optimizer, data_module.vocabulary)

    train(code2class, data_module, config)


def test_code2class(config: DictConfig):