IndexToString extends MapModel<IndexToString> implements IndexToStringPredictParams<IndexToString> { public IndexToString() { this(new Params()); } IndexToString();  IndexToString(Params params); @Override Table getModelData();  }
@Test public void testIndexToString() throws Exception { BatchOperator data = new MemSourceBatchOp(Arrays.asList(rows), new String[]{"f0"}); StringIndexer stringIndexer = new StringIndexer() .setModelName("string_indexer_model") .setSelectedCol("f0") .setOutputCol("f0_indexed") .setStringOrderType("frequency_asc"); BatchOperator indexed = stringIndexer.fit(data).transform(data); IndexToString indexToString = new IndexToString() .setModelName("string_indexer_model") .setSelectedCol("f0_indexed") .setOutputCol("f0_indxed_unindexed"); List<Row> unindexed = indexToString.transform(indexed).collect(); unindexed.forEach(row -> { Assert.assertEquals(row.getField(0), row.getField(2)); }); }