Once your model is trained, you also have to store GlobalCounter for good, is that correct? I would like the stores to be as specific as possible, hence my question: is it possible to use ContextStore for entityword and CandidateMapStore for entitymention? topicentity would then be implemented as a third store, e.g. TopicEntityStore.