コード例 #1
0
ファイル: TFIDF.java プロジェクト: Edunov/sparsity
 @Override
 public BasicMatrix train(BasicMatrix input) throws Exception {
   input.map(
       new MatrixMapper() {
         @Override
         public void map(int row, int col, float value) {
           docCounts.adjustOrPutValue(col, 1, 1);
         }
       });
   N = input.rows();
   return process(input);
 }
コード例 #2
0
ファイル: TFIDF.java プロジェクト: Edunov/sparsity
 @Override
 public BasicMatrix process(final BasicMatrix input) throws Exception {
   input.map(
       new MatrixMapper() {
         @Override
         public void map(int row, int col, float value) {
           input.set(row, col, tf((int) value) * idf(col));
         }
       });
   return input;
 }