:robot: Natural language processing with JavaScript
npm i classifier.js
# or
yarn add classifier.js
import { Classifier } from 'classifier.js'
const classifier = new Classifier({ returnType: 'PERCENTAGE' })
classifier.learn('I like cats', ['animal'])
classifier.learn('Cats are cool', ['animal'])
classifier.learn('Dogs are noisy', ['animal'])
classifier.learn('I love animals', ['animal'])
classifier.learn('I like my horse', ['animal'])
classifier.learn('Chocolate is good', ['food'])
classifier.learn('I eat apple', ['food'])
classifier.learn('Juice is very good', ['food'])
classifier.learn('Brazilians eat rice and beans', ['food'])
classifier.learn('Bananas are good for health', ['food'])
classifier.classify('Apple juice is awesome')
// OUTPUT: { unknown: '20%', animal: '0%', food: '80%' }
import { Classifier } from 'classifier.js'
const classifier = new Classifier({ percentualReturn: true })
classifier.learn('000.000.000-11', ['cpf'])
classifier.learn('00.000.000/0001-00', ['cnpj'])
classifier.learn('00155-333', ['zipcode'])
classifier.classify('999.999.999-99')
// OUTPUT: { unknown: '0%', cpf: '100%', cnpj: '0%', zipcode: '0%' }
classifier.classify('99.999.999/9999-99')
// OUTPUT: { unknown: '0%', cpf: '0%', cnpj: '100%', zipcode: '0%' }
classifier.classify('99999-999')
// OUTPUT: { unknown: '0%', cpf: '0%', cnpj: '0%', zipcode: '100%' }
learn
Receive data with an array of related categories.
classifier.learn('Your sentence', ['your-category'])
classify
Classify a sentence based on received data.
classifier.classify('Sentence to classify')
resetKnowledge
Removes all that was learned.
classifier.resetKnowledge()
toJSON
Saves classifier data to a JSON file that can be imported later.
classifier.toJSON('myFolder/savedClassifier.json')
# Or simply
classifier.toJSON('savedClassifier.json')
fromJSON
Imports data from a JSON file.
classifier.fromJSON('myFolder/savedClassifier.json')
# Or simply
classifier.fromJSON('savedClassifier.json')
toYAML
Saves classifier data to an YAML file that can be imported later.
classifier.toYAML('myFolder/savedClassifier.yaml')
# Or simply
classifier.toYAML('savedClassifier.yaml')
fromYAML
Imports data from an YAML file.
classifier.fromYAML('myFolder/savedClassifier.yaml')
# Or simply
classifier.fromYAML('savedClassifier.yaml')