Examples with Transform water.rapids.transforms.Transform used on opensource projects

Example 1 with Transform

use of water.rapids.transforms.Transform in project h2o-3 by h2oai.

the class Assembly method toJava.

public String toJava(String pojoName) {
    if (pojoName == null)
        pojoName = "GeneratedMungingPojo";
    StringBuilder sb = new StringBuilder("import hex.genmodel.GenMunger;\n" + "import hex.genmodel.easy.RowData;\n\n" + "public class " + pojoName + " extends GenMunger {\n" + "  public " + pojoName + "() {\n" + "    _steps = new Step[" + _steps.length + "];\n");
    int i = 0;
    for (Transform step : _steps) sb.append("    _steps[").append(i++).append("] = new ").append(step.name()).append("();\n");
    sb.append("  }\n");
    for (Transform step : _steps) sb.append(step.genClass());
    sb.append("}\n");
    return sb.toString();
}

Example 2 with Transform

use of water.rapids.transforms.Transform in project h2o-3 by h2oai.

the class AssemblyHandler method fit.

public AssemblyV99 fit(int version, AssemblyV99 ass) throws ClassNotFoundException, NoSuchMethodException, IllegalAccessException, InvocationTargetException, InstantiationException {
    if (ass == null)
        return null;
    if (ass.steps == null)
        return ass;
    // process assembly:
    //   of the form [name__class__ast__inplace__names, name__class__ast__inplace__names, ...]
    // s[0] : stepName
    // s[1] : transform class
    // s[2] : ast (can be noop)
    // s[3] : inplace
    // s[4] : names
    ArrayList<Transform> steps = new ArrayList<>();
    for (String step : ass.steps) {
        String[] s = step.split("__");
        Class transformClass = Class.forName("water.rapids.transforms." + s[1]);
        Class[] constructorTypes = new Class[] { String.class, /*name*/
        String.class, /*ast*/
        boolean.class, /*inplace*/
        String[].class };
        Object[] constructorArgs = new Object[] { s[0], s[2], Boolean.valueOf(s[3]), s[4].equals("|") ? null : s[4].split("\\|") };
        steps.add((Transform) transformClass.getConstructor(constructorTypes).newInstance(constructorArgs));
    }
    Assembly assembly = new Assembly(Key.make("assembly_" + Key.make().toString()), steps.toArray(new Transform[steps.size()]));
    ass.result = new KeyV3.FrameKeyV3(assembly.fit((Frame) DKV.getGet(ass.frame.key()))._key);
    ass.assembly = new KeyV3.AssemblyKeyV3(assembly._key);
    DKV.put(assembly);
    return ass;
}