UNPKG

dbrans-natural

Version:

General natural language (tokenizing, stemming (English, Russian, Spanish), classification, inflection, phonetics, tfidf, WordNet, jaro-winkler, Levenshtein distance, Dice's Coefficient) facilities for node.

710 lines (515 loc) 20.2 kB
natural ======= "Natural" is a general natural language facility for node.js. Tokenizing, stemming, classification, phonetics, tf-idf, WordNet, string similarity, and some inflections are currently supported. It's still in the early stages, so we're very interested in bug reports, contributions and the like. Note that many algorithms from Rob Ellis's [node-nltools](https://github.com/NaturalNode/node-nltools) are being merged into this project and will be maintained from here onward. At the moment, most of the algorithms are English-specific, but in the long-term, some diversity will be in order. Thanks to Polyakov Vladimir, Russian stemming has been added!, Thanks to David Przybilla, Spanish stemming has been added!. Aside from this README, the only documentation is [this DZone article](http://www.dzone.com/links/r/using_natural_a_nlp_module_for_nodejs.html) and [here on my blog](http://www.chrisumbel.com/article/node_js_natural_language_porter_stemmer_lancaster_bayes_naive_metaphone_soundex), which is a bit older. Looking for Help Maintaining Natural! ------------------------------------- I'm having trouble devoting the time necessary to maintain natural. While I'm certainly not leaving the project I'd like someone to take over the day-to-day maintenance of dealing with issues, pull requests and driving the direction of the software moving forward. Please contact chris@chrisumbel.com if you're interested! Installation ------------ If you're just looking to use natural without your own node application, you can install via NPM like so: npm install natural If you're interested in contributing to natural, or just hacking on it, then by all means fork away! Tokenizers ---------- Word, Regexp, and Treebank tokenizers are provided for breaking text up into arrays of tokens: ```javascript var natural = require('natural'), tokenizer = new natural.WordTokenizer(); console.log(tokenizer.tokenize("your dog has flees.")); // [ 'your', 'dog', 'has', 'flees' ] ``` The other tokenizers follow a similar pattern: ```javascript tokenizer = new natural.TreebankWordTokenizer(); console.log(tokenizer.tokenize("my dog hasn't any flees.")); // [ 'my', 'dog', 'has', 'n\'t', 'any', 'flees', '.' ] tokenizer = new natural.RegexpTokenizer({pattern: /\-/}); console.log(tokenizer.tokenize("flee-dog")); // [ 'flee', 'dog' ] tokenizer = new natural.WordPunctTokenizer(); console.log(tokenizer.tokenize("my dog hasn't any flees.")); // [ 'my', 'dog', 'hasn', '\'', 't', 'any', 'flees', '.' ] ``` String Distance ---------------------- Natural provides an implementation of the Jaro-Winkler string distance measuring algorithm. This will return a number between 0 and 1 which tells how closely the strings match (0 = not at all, 1 = exact match): ```javascript var natural = require('natural'); console.log(natural.JaroWinklerDistance("dixon","dicksonx")) console.log(natural.JaroWinklerDistance('not', 'same')); ``` Output: ```javascript 0.7466666666666666 0 ``` Natural also offers support for Levenshtein distances: ```javascript var natural = require('natural'); console.log(natural.LevenshteinDistance("ones","onez")); console.log(natural.LevenshteinDistance('one', 'one')); ``` Output: ```javascript 1 0 ``` The cost of the three edit operations are modifiable for Levenshtein: ```javascript console.log(natural.LevenshteinDistance("ones","onez", { insertion_cost: 1, deletion_cost: 1, substitution_cost: 1 })); ``` Output: ```javascript 1 ``` And Dice's co-efficient: ```javascript var natural = require('natural'); console.log(natural.DiceCoefficient('thing', 'thing')); console.log(natural.DiceCoefficient('not', 'same')); ``` Output: ```javascript 1 0 ``` Stemmers -------- Currently, stemming is supported via the Porter (English,Russian and Spanish) and Lancaster (Paice/Husk) algorithms. ```javascript var natural = require('natural'); ``` This example uses a Porter stemmer. "word" is returned. ```javascript console.log(natural.PorterStemmer.stem("words")); // stem a single word ``` in Russian: ```javascript console.log(natural.PorterStemmerRu.stem("падший")); ``` in Spanish: ```javascript console.log(natural.PorterStemmerEs.stem("jugaría")); ``` `attach()` patches `stem()` and `tokenizeAndStem()` to String as a shortcut to `PorterStemmer.stem(token)`. `tokenizeAndStem()` breaks text up into single words and returns an array of stemmed tokens. ```javascript natural.PorterStemmer.attach(); console.log("i am waking up to the sounds of chainsaws".tokenizeAndStem()); console.log("chainsaws".stem()); ``` the same thing can be done with a Lancaster stemmer: ```javascript natural.LancasterStemmer.attach(); console.log("i am waking up to the sounds of chainsaws".tokenizeAndStem()); console.log("chainsaws".stem()); ``` Classifiers ---------------------- Two classifiers are currently supported, Naive Bayes and logistic regression. The following examples use the BayesClassifier class, but the LogisticRegressionClassifier class could be substituted instead. ```javascript var natural = require('natural'), classifier = new natural.BayesClassifier(); ``` You can train the classifier on sample text. It will use reasonable defaults to tokenize and stem the text. ```javascript classifier.addDocument('i am long qqqq', 'buy'); classifier.addDocument('buy the q''s', 'buy'); classifier.addDocument('short gold', 'sell'); classifier.addDocument('sell gold', 'sell'); classifier.train(); ``` Outputs "sell" ```javascript console.log(classifier.classify('i am short silver')); ``` Outputs "buy" ```javascript console.log(classifier.classify('i am long copper')); ``` You have access to the set of matched classes and the associated value from the classifier. Outputs: ```javascript [ { label: 'sell', value: 0.39999999999999997 }, { label: 'buy', value: 0.19999999999999998 } ] ``` From this: ```javascript console.log(classifier.getClassifications('i am long copper')); ``` The classifier can also be trained with and can classify arrays of tokens, strings, or any mixture of the two. Arrays let you use entirely custom data with your own tokenization/stemming, if you choose to implement it. ```javascript classifier.addDocument(['sell', 'gold'], 'sell'); ``` A classifier can also be persisted and recalled later so that you can reuse it later. ```javascript classifier.save('classifier.json', function(err, classifier) { // the classifier is saved to the classifier.json file! }); ``` To recall from the classifier.json saved above: ```javascript natural.BayesClassifier.load('classifier.json', null, function(err, classifier) { console.log(classifier.classify('long SUNW')); console.log(classifier.classify('short SUNW')); }); ``` A classifier can also be serialized and deserialized like so: ```javascript var classifier = new natural.BayesClassifier(); classifier.addDocument(['sell', 'gold'], 'sell'); classifier.addDocument(['buy', 'silver'], 'buy'); // serialize var raw = JSON.stringify(classifier); // deserialize var restoredClassifier = natural.BayesClassifier.restore(JSON.parse(raw)); console.log(restoredClassifier.classify('i should sell that')); ``` Phonetics --------- Phonetic matching (sounds-like) matching can be done withthe SoundEx, Metaphone, or DoubleMetaphone algorithms: ```javascript var natural = require('natural'), metaphone = natural.Metaphone, soundEx = natural.SoundEx; var wordA = 'phonetics'; var wordB = 'fonetix'; ``` To test the two words to see if they sound alike: ```javascript if(metaphone.compare(wordA, wordB)) console.log('they sound alike!'); ``` The raw phonetics are obtained with `process()`: ```javascript console.log(metaphone.process('phonetics')); ``` A maximum code length can be supplied: ```javascript console.log(metaphone.process('phonetics', 3)); ``` `DoubleMetaphone` deals with two encodings returned in an array. This feature is experimental and subject to change: ```javascript var natural = require('natural'), dm = natural.DoubleMetaphone; var encodings = dm.process('Matrix'); console.log(encodings[0]); console.log(encodings[1]); ``` Attaching will patch String with useful methods: ```javascript metaphone.attach(); ``` `soundsLike` is essentially a shortcut to `Metaphone.compare`: ```javascript if(wordA.soundsLike(wordB)) console.log('they sound alike!'); ``` The raw phonetics are obtained with `phonetics()`: ```javascript console.log('phonetics'.phonetics()); ``` Full text strings can be tokenized into arrays of phonetics (much like how tokenization-to-arrays works for stemmers): ```javascript console.log('phonetics rock'.tokenizeAndPhoneticize()); ``` Same module operations applied with `SoundEx`: ```javascript if(soundEx.compare(wordA, wordB)) console.log('they sound alike!'); ``` The same String patches apply with `soundEx`: ```javascript soundEx.attach(); if(wordA.soundsLike(wordB)) console.log('they sound alike!'); console.log('phonetics'.phonetics()); ``` Inflectors ---------- ### Nouns Nouns can be pluralized/singularized with a `NounInflector`: ```javascript var natural = require('natural'), nounInflector = new natural.NounInflector(); ``` To pluralize a word (outputs "radii"): ```javascript console.log(nounInflector.pluralize('radius')); ``` To singularize a word (outputs "beer"): ```javascript console.log(nounInflector.singularize('beers')); ``` Like many of the other features, String can be patched to perform the operations directly. The "Noun" suffix on the methods is necessary, as verbs will be supported in the future. ```javascript nounInflector.attach(); console.log('radius'.pluralizeNoun()); console.log('beers'.singularizeNoun()); ``` ### Numbers Numbers can be counted with a CountInflector: ```javascript var countInflector = natural.CountInflector; ``` Outputs "1st": ```javascript console.log(countInflector.nth(1)); ``` Outputs "111th": ```javascript console.log(countInflector.nth(111)); ``` ### Present Tense Verbs Present Tense Verbs can be pluralized/singularized with a PresentVerbInflector. This feature is still experimental as of 0.0.42, so use with caution, and please provide feedback. ```javascript var verbInflector = new natural.PresentVerbInflector(); ``` Outputs "becomes": ```javascript console.log(verbInflector.singularize('become')); ``` Outputs "become": ```javascript console.log(verbInflector.pluralize('becomes')); ``` Like many other natural modules, `attach()` can be used to patch strings with handy methods. ```javascript verbInflector.attach(); console.log('walk'.singularizePresentVerb()); console.log('walks'.pluralizePresentVerb()); ``` N-Grams ------- n-grams can be obtained for either arrays or strings (which will be tokenized for you): ```javascript var NGrams = natural.NGrams; ``` ### bigrams ```javascript console.log(NGrams.bigrams('some words here')); console.log(NGrams.bigrams(['some', 'words', 'here'])); ``` Both of the above output: `[ [ 'some', 'words' ], [ 'words', 'here' ] ]` ### trigrams ```javascript console.log(NGrams.trigrams('some other words here')); console.log(NGrams.trigrams(['some', 'other', 'words', 'here'])); ``` Both of the above output: `[ [ 'some', 'other', 'words' ], [ 'other', 'words', 'here' ] ]` ### arbitrary n-grams ```javascript console.log(NGrams.ngrams('some other words here for you', 4)); console.log(NGrams.ngrams(['some', 'other', 'words', 'here', 'for', 'you'], 4)); ``` The above outputs: `[ [ 'some', 'other', 'words', 'here' ], [ 'other', 'words', 'here', 'for' ], [ 'words', 'here', 'for', 'you' ] ]` tf-idf ----- Term Frequency–Inverse Document Frequency (tf-idf) is implemented to determine how important a word (or words) is to a document relative to a corpus. The following example will add four documents to a corpus and determine the weight of the word "node", then the weight of the word "ruby" in each document. ```javascript var natural = require('natural'), TfIdf = natural.TfIdf, tfidf = new TfIdf(); tfidf.addDocument('this document is about node.'); tfidf.addDocument('this document is about ruby.'); tfidf.addDocument('this document is about ruby and node.'); tfidf.addDocument('this document is about node. it has node examples'); console.log('node --------------------------------'); tfidf.tfidfs('node', function(i, measure) { console.log('document #' + i + ' is ' + measure); }); console.log('ruby --------------------------------'); tfidf.tfidfs('ruby', function(i, measure) { console.log('document #' + i + ' is ' + measure); }); ``` The above outputs: ``` node -------------------------------- document #0 is 1.4469189829363254 document #1 is 0 document #2 is 1.4469189829363254 document #3 is 2.8938379658726507 ruby -------------------------------- document #0 is 0 document #1 is 1.466337068793427 document #2 is 1.466337068793427 document #3 is 0 ``` This approach can also be applied to individual documents. The following example measures the term "node" in the first and second documents. ```javascript console.log(tfidf.tfidf('node', 0)); console.log(tfidf.tfidf('node', 1)); ``` A TfIdf instance can also load documents from files on disk. ```javascript var tfidf = new TfIdf(); tfidf.addFileSync('data_files/one.txt'); tfidf.addFileSync('data_files/two.txt'); ``` Multiple terms can be measured as well, with their weights being added into a single measure value. The following example determines that the last document is the most relevent to the words "node" and "ruby". ```javascript var natural = require('natural'), TfIdf = natural.TfIdf, tfidf = new TfIdf(); tfidf.addDocument('this document is about node.'); tfidf.addDocument('this document is about ruby.'); tfidf.addDocument('this document is about ruby and node.'); tfidf.tfidfs('node ruby', function(i, measure) { console.log('document #' + i + ' is ' + measure); }); ``` The above outputs: ``` document #0 is 1.2039728043259361 document #1 is 1.2039728043259361 document #2 is 2.4079456086518722 ``` The examples above all use strings, which case natural to automatically tokenize the input. If you wish to perform your own tokenization or other kinds of processing, you can do so, then pass in the resultant arrays later. This approach allows you to bypass natural's default preprocessing. ```javascript var natural = require('natural'), TfIdf = natural.TfIdf, tfidf = new TfIdf(); tfidf.addDocument(['document', 'about', 'node']); tfidf.addDocument(['document', 'about', 'ruby']); tfidf.addDocument(['document', 'about', 'ruby', 'node']); tfidf.addDocument(['document', 'about', 'node', 'node', 'examples']); tfidf.tfidfs(['node', 'ruby'], function(i, measure) { console.log('document #' + i + ' is ' + measure); }); ``` It's possible to retrieve a list of all terms in a document, sorted by their importance. ```javascript tfidf.listTerms(0 /*document index*/).forEach(function(item) { console.log(item.term + ': ' + item.tfidf); }); ``` A TfIdf instance can also be serialized and deserialzed for save and recall. ```javascript var tfidf = new TfIdf(); tfidf.addDocument('document one', 'un'); tfidf.addDocument('document Two', 'deux'); var s = JSON.stringify(tfidf); // save "s" to disk, database or otherwise // assuming you pulled "s" back out of storage. var tfidf = new TfIdf(JSON.parse(s)); ``` WordNet ------- One of the newest and most experimental features in natural is WordNet integration. Here's an example of using natural to look up definitions of the word node. To use the WordNet module, first install the WordNet database files using the [WNdb module](https://github.com/moos/WNdb): npm install WNdb (For node < v0.6, please use 'npm install WNdb@3.0.0') Keep in mind that the WordNet integration is to be considered experimental at this point, and not production-ready. The API is also subject to change. Here's an exmple of looking up definitions for the word, "node". ```javascript var wordnet = new natural.WordNet(); wordnet.lookup('node', function(results) { results.forEach(function(result) { console.log('------------------------------------'); console.log(result.synsetOffset); console.log(result.pos); console.log(result.lemma); console.log(result.synonyms); console.log(result.pos); console.log(result.gloss); }); }); ``` Given a synset offset and a part of speech, a definition can be looked up directly. ```javascript var wordnet = new natural.WordNet(); wordnet.get(4424418, 'n', function(result) { console.log('------------------------------------'); console.log(result.lemma); console.log(result.pos); console.log(result.gloss); console.log(result.synonyms); }); ``` If you have _manually_ downloaded the WordNet database files, you can pass the folder to the constructor: ```javascript var wordnet = new natural.WordNet('/my/wordnet/dict'); ``` As of v0.1.11, WordNet data files are no longer automatically downloaded. Princeton University "About WordNet." WordNet. Princeton University. 2010. <http://wordnet.princeton.edu> Development ----------- When developing, please: + Write unit tests + Make sure your unit tests pass The current configuration of the unit tests requires the following environment variable to be set: export NODE_PATH=. License ------- Copyright (c) 2011, 2012 Chris Umbel, Rob Ellis, Russell Mull Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. WordNet License --------------- This license is available as the file LICENSE in any downloaded version of WordNet. WordNet 3.0 license: (Download) WordNet Release 3.0 This software and database is being provided to you, the LICENSEE, by Princeton University under the following license. By obtaining, using and/or copying this software and database, you agree that you have read, understood, and will comply with these terms and conditions.: Permission to use, copy, modify and distribute this software and database and its documentation for any purpose and without fee or royalty is hereby granted, provided that you agree to comply with the following copyright notice and statements, including the disclaimer, and that the same appear on ALL copies of the software, database and documentation, including modifications that you make for internal use or for distribution. WordNet 3.0 Copyright 2006 by Princeton University. All rights reserved. THIS SOFTWARE AND DATABASE IS PROVIDED "AS IS" AND PRINCETON UNIVERSITY MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, PRINCETON UNIVERSITY MAKES NO REPRESENTATIONS OR WARRANTIES OF MERCHANT- ABILITY OR FITNESS FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF THE LICENSED SOFTWARE, DATABASE OR DOCUMENTATION WILL NOT INFRINGE ANY THIRD PARTY PATENTS, COPYRIGHTS, TRADEMARKS OR OTHER RIGHTS. The name of Princeton University or Princeton may not be used in advertising or publicity pertaining to distribution of the software and/or database. Title to copyright in this software, database and any associated documentation shall at all times remain with Princeton University and LICENSEE agrees to preserve same.