Word2VecModelMapper extends SISOModelMapper { @Override protected Object predictResult(Object input) throws Exception { return generator.transform((String) input); } Word2VecModelMapper(TableSchema modelSchema, TableSchema dataSchema, Params params); @Override void loadModel(List <Row> modelRows);  }
@Test public void predictResult() { String type = "avg"; assertEquals( HasPredMethod.PredMethod.valueOf(type.trim().toUpperCase()), HasPredMethod.PredMethod.AVG); }