public static void Train(XMLToAnnotation.Data data, Properties props) { IDictionary <string, IList <Person> > characterMap = QuoteAttributionUtils.ReadPersonMap(props.GetProperty("charactersPath")); IDictionary <int, string> pronounCorefMap = QuoteAttributionUtils.SetupCoref(props.GetProperty("booknlpCoref"), characterMap, data.doc); ICollection <string> animacyList = QuoteAttributionUtils.ReadAnimacyList(QuoteAttributionAnnotator.AnimacyWordList); SupervisedSieveTraining.FeaturesData fd = Featurize(new SupervisedSieveTraining.SieveData(data.doc, characterMap, pronounCorefMap, animacyList), data.goldList, true); ExtractQuotesClassifier quotesClassifier = new ExtractQuotesClassifier(fd.dataset); OutputModel(props.GetProperty("modelPath"), quotesClassifier.GetClassifier()); }
public virtual void LoadModel(string filename) { quotesClassifier = new ExtractQuotesClassifier(filename); }