Question

Simple code using Java to create a program that uses TF-IDF algorithm. Please add step by step on how to run the code.

Answer 1

Main.java

import java.util.Arrays;

import java.util.List;

public class Main {

public double frequency(List<String> lt, String str) {

double res = 0;

for (String str1 : lt) {

if (str.equalsIgnoreCase(str1))

res++;

}

return res / lt.size();

}

public double idf(List<List<String>> lt1, String str) {

double num = 0;

for (List<String> lt : lt1) {

for (String str1 : lt) {

if (str.equalsIgnoreCase(str1)) {

num++;

break;

}

}

}

return Math.log(lt1.size() / num);

}

public double frequencyDec(List<String> lt, List<List<String>> lt1, String str) {

return frequency(lt, str) * idf(lt1, str);

}

public static void main(String[] args) {

List<String> lt2 = Arrays.asList("Yo", "Hello", "Bro", "Sissy", "sit", "Nte");

List<String> lt3 = Arrays.asList("vit", "te", "at", "Sissy", "pro", "quo");

List<String> lt4 = Arrays.asList("Has", "persius", "diss", "dis", "simulation");

List<List<String>> lt5 = Arrays.asList(lt2, lt3, lt4);

Main m = new Main();

double freq1 = m.frequencyDec(lt2, lt5, "Sissy");

System.out.println("TF-IDF (Sissy) = " + freq1);

}

}

Output: