presspage_tfidf_document_corpus_term_frequency_calculation.ts

1import { TfIdf } from '@presspage/tf-idf';
2
3const tfidf = new TfIdf();
4
5// Add documents to the corpus
6tfidf.addDocument('this document is about node.js');
7tfidf.addDocument('this document is about ruby');
8tfidf.addDocument('this document is about ruby and node.js');
9tfidf.addDocument('this document is about node.js. node.js is great');
10
11// Calculate TF-IDF for the term 'node.js' in the first document (index 0)
12console.log('node.js (doc 1): ', tfidf.tfidf('node.js', 0));
13
14// Calculate TF-IDF for the term 'ruby' in the second document (index 1)
15console.log('ruby (doc 2): ', tfidf.tfidf('ruby', 1));
16
17// List measures for all documents for a specific term
18tfidf.listTerms(0).forEach((item) => {
19    console.log(`${item.term}: ${item.tfidf}`);
20});