TF-IDF计算四

最后, 开始计算TF-IDF

TF i - IDF i,j = TF i,j * IDF i

代码:

class TFIDF

{

ArrayList<HashMap<String,Double>> TFIDFMainFileList = new ArrayList<HashMap<String,Double>>();

ArrayList<HashMap<String,Double>> TFMainFileList = new ArrayList<HashMap<String,Double>>();

ArrayList<ArrayList<String>> MainFileList = new ArrayList<ArrayList<String>>();

HashMap<String,Double> IDFMainFileList = new HashMap<String, Double>();

public TFIDF(ArrayList<ArrayList<String>> mfl, HashMap<String,Double> idfm, ArrayList<HashMap<String,Double>> tfmfl)

{

MainFileList = mfl;

IDFMainFileList = idfm;

TFMainFileList = tfmfl;

}

public ArrayList<HashMap<String,Double>> PrintTFIDF()

{

for(int i=0; i<MainFileList.size(); i++)

{

ArrayList<String> SubFileList = MainFileList.get(i);

HashMap<String,Double> tfFile = TFMainFileList.get(i);

HashMap<String,Double> GetTFIDF = new HashMap<String, Double>();

ArrayList<String> Index = new ArrayList<String>();

//Take hashmap level from ArrayList<HashMap<String,Double>>

for(int j=0; j<SubFileList.size(); j++)

{

Index.add(SubFileList.get(j));

double tf = tfFile.get(SubFileList.get(j));

double idf = IDFMainFileList.get(SubFileList.get(j));

double tfidf = tf * idf;

GetTFIDF.put(SubFileList.get(j),tfidf);

}

TFIDFMainFileList.add(GetTFIDF);

//==========================================================

}

return TFIDFMainFileList;

}

}