Open Source Love version PRs Welcome CD
π€ An library for natural language processing with JavaScript
npm i classifier.js
# or
yarn add classifier.jsimport { 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%' }
Receive data with an array of related categories.
classifier.learn('Your sentence', ['your-category'])
Classify a sentence based on received data.
classifier.classify('Sentence to classify')
Removes all that was learned.
classifier.resetKnowledge()
Saves classifier data to a JSON file that can be imported later.
classifier.toJSON('myFolder/savedClassifier.json') # Or simply classifier.toJSON('savedClassifier.json')
Imports data from a JSON file.
classifier.fromJSON('myFolder/savedClassifier.json') # Or simply classifier.fromJSON('savedClassifier.json')
Saves classifier data to an YAML file that can be imported later.
classifier.toYAML('myFolder/savedClassifier.yaml') # Or simply classifier.toYAML('savedClassifier.yaml')
Imports data from an YAML file.
classifier.fromYAML('myFolder/savedClassifier.yaml') # Or simply classifier.fromYAML('savedClassifier.yaml')