TfidfVectorizer extends org.canova.nlp.vectorizer.TfidfVectorizer<INDArray> { @Override public INDArray fitTransform(RecordReader reader) { return fitTransform(reader,null); }  @Override INDArray createVector(Object[] args); @Override INDArray fitTransform(RecordReader reader); @Override INDArray fitTransform(RecordReader reader, RecordCallBack callBack); @Override INDArray transform(Collection<Writable> record);  }
@Test public void testTfidfVectorizer() { TfidfVectorizer vectorizer = new TfidfVectorizer(); vectorizer.initialize(new Configuration()); RecordReader reader = new CollectionRecordReader(Writables.writables(Arrays.asList("Testing one.", "Testing 2."))); INDArray n = vectorizer.fitTransform(reader); assertEquals(3,n.columns()); assertEquals(2,n.rows()); }