libcity.executor.gensim_executor¶

class libcity.executor.gensim_executor.GensimExecutor(config, model, data_feature)[source]¶

Bases: libcity.executor.abstract_executor.AbstractExecutor

evaluate(test_dataloader)[source]¶: use model to test data

load_model(cache_name)[source]¶

加载对应模型的 cache

Parameters: cache_name (str) – 保存的文件名

save_model(cache_name)[source]¶

将当前的模型保存到文件

Parameters: cache_name (str) – 保存的文件名

train(train_dataloader, eval_dataloader)[source]¶: use data to train model with config