Skip to content

Instantly share code, notes, and snippets.

@duanebester
Created February 11, 2020 19:39
Show Gist options
  • Save duanebester/d30531137885c2b2533eb877c6e38c9f to your computer and use it in GitHub Desktop.
Save duanebester/d30531137885c2b2533eb877c6e38c9f to your computer and use it in GitHub Desktop.
NLP Trait
trait NLP {
lazy val tokenModel = new TokenizerModel(getClass.getResourceAsStream("/en-token.bin"))
lazy val tokenizer = new TokenizerME(tokenModel);
lazy val sentenceModel = new SentenceModel(getClass.getResourceAsStream("/en-sent.bin"))
lazy val sentenceDetector = new SentenceDetectorME(sentenceModel);
lazy val personModel = new TokenNameFinderModel(getClass.getResourceAsStream("/en-ner-person.bin"))
lazy val personFinderME = new NameFinderME(personModel);
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment