libcity.executor.abstract_tradition_executor

class libcity.executor.abstract_tradition_executor.AbstractTraditionExecutor(config, model, data_feature)[source]

Bases: libcity.executor.abstract_executor.AbstractExecutor

evaluate(test_dataloader)[source]

use model to test data

Parameters

test_dataloader (torch.Dataloader) – Dataloader

load_model(cache_name)[source]

对于传统模型,不需要模型加载

Parameters

cache_name (str) – 保存的文件名

save_model(cache_name)[source]

对于传统模型,不需要模型保存

Parameters

cache_name (str) – 保存的文件名

train(train_dataloader, eval_dataloader)[source]

train model

Parameters
  • train_dataloader (torch.Dataloader) – Dataloader

  • eval_dataloader (torch.Dataloader) – Dataloader