https://github.com/pharo-ai/TF-IDF

Implementation of TF-IDF in Pharo
https://github.com/pharo-ai/TF-IDF

pharo statistics term-frequency tf-idf

Last synced: about 2 months ago
JSON representation

Implementation of TF-IDF in Pharo

• Host: GitHub
• URL: https://github.com/pharo-ai/TF-IDF
• Owner: pharo-ai
• License: mit
• Created: 2019-04-17T14:51:07.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2022-02-04T10:32:01.000Z (almost 3 years ago)
• Last Synced: 2024-08-03T17:08:56.557Z (5 months ago)
• Topics: pharo, statistics, term-frequency, tf-idf
• Language: Smalltalk
• Size: 38.1 KB
• Stars: 3
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-pharo - pharo-ai / TF-IDF - Term Frequency - Inverse Document Frequency (TF-IDF), a statistical metric that reflects the importance of a word in a document. Can be used for finding keywords, ranking words by importance, or as a simple way of finding semantic similarity between documents. (Artificial Intelligence and Machine Learning)

README

        
# Term Frequency - Inverse Document Frequency (TF-IDF)

[![Build status](https://github.com/pharo-ai/tf-idf/workflows/CI/badge.svg)](https://github.com/pharo-ai/tf-idf/actions/workflows/test.yml)

[![Coverage Status](https://coveralls.io/repos/github/pharo-ai/TF-IDF/badge.svg?branch=master)](https://coveralls.io/github/pharo-ai/TF-IDF?branch=master)

[![License](https://img.shields.io/badge/license-MIT-blue.svg)](https://raw.githubusercontent.com/pharo-ai/TF-IDF/master/LICENSE)

This repository contains the implementation of TF-IDF algorithm in Pharo.

For more infomation please refer to the Pharo-AI wiki: https://github.com/pharo-ai/wiki

## How to install it

To install `TF-IDF`, go to the Playground (Ctrl+OW) in your [Pharo](https://pharo.org/) image and execute the following Metacello script (select it and press Do-it button or Ctrl+D):

```Smalltalk

Metacello new

  baseline: 'AITfIdf';

  repository: 'github://pharo-ai/tf-idf/src';

  load.

```

## How to depend on it

If you want to add a dependency on `TF-IDF` to your project, include the following lines into your baseline method:

```Smalltalk

spec

  baseline: 'AITfIdf'

  with: [ spec repository: 'github://pharo-ai/tf-idf/src' ].

```

If you are new to baselines and Metacello, check out the [Baselines](https://github.com/pharo-open-documentation/pharo-wiki/blob/master/General/Baselines.md) tutorial on Pharo Wiki.

## How to use it

Here is a simple example of how you can train a TF-IDF model and use it to assign scores to words. You are given an array of sentences where each sentence is represented as an array of words:

```Smalltalk

sentences := #(

  (I am Sam)

  (Sam I am)

  (I 'don''t' like green eggs and ham)).

```

Train a TF-IDF model on those sentences:

```Smalltalk

tfidf := AITermFrequencyInverseDocumentFrequency new.

tfidf trainOn: sentences.

```

Use it to assign TF-IDF scores to words:

```Smalltalk

tfidf scoreOf: 'Sam' in: #(I am Sam). "0.4054651081081644"

```

You can also encode any given text with a TF-IDF vector

```Smalltalk

tfidf vectorFor: #(I am green green ham). "#(0.0 0.0 0.4054651081081644 0.0 0.0 0.0 2.1972245773362196 1.0986122886681098 0.0)"

```