libcity.executor.abstract_executor

class libcity.executor.abstract_executor.AbstractExecutor(config, model, data_feature)

Bases: object

evaluate(test_dataloader)

use model to test data

Parameters

test_dataloader (torch.Dataloader) – Dataloader

load_model(cache_name)

加载对应模型的 cache

Parameters

cache_name (str) – 保存的文件名

save_model(cache_name)

将当前的模型保存到文件

Parameters

cache_name (str) – 保存的文件名

train(train_dataloader, eval_dataloader)

use data to train model with config

Parameters

• train_dataloader (torch.Dataloader) – Dataloader

• eval_dataloader (torch.Dataloader) – Dataloader