libcity.executor.gensim_executor

class libcity.executor.gensim_executor.GensimExecutor(config, model, data_feature)

Bases: libcity.executor.abstract_executor.AbstractExecutor

evaluate(test_dataloader)

use model to test data

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