def train_types(self):
"""
Train models for each subtype of entity, and a general model.
:param types: subtypes of entities to train individual models, as well as a general model
"""
self.basemodel.load_data(self.corpus,
feature_extractors.keys(),
subtype="all")
for t in self.types:
typepath = self.basepath + "_" + t
model = StanfordNERModel(typepath,
subtypes=self.basemodel.subtypes)
model.copy_data(self.basemodel, t)
logging.info("training subtype %s" % t)
model.train()
self.models[t] = model
