private String[] getArgAsStrings(Env env, Env.StackHelp stk, AstRoot ast) { String[] frameKeys; if (ast instanceof AstStrList) { AstStrList teColumns = ((AstStrList) ast); frameKeys = teColumns._strs; } else if( ast instanceof AstStr) { String teColumn = stk.track(ast.exec(env)).getStr(); frameKeys = new String[]{teColumn}; } else { throw new IllegalStateException("Failed to parse ast parameter: " + ast.toString()); } return frameKeys; }
@Test public void calculateSingleNumberResultTest() { fr = new TestFrameBuilder() .withName("testFrame") .withColNames("ColA") .withVecTypes(Vec.T_NUM) .withDataForCol(0, ard(1, 2, 3)) .build(); String tree = "(sum (cols testFrame [0.0] ))"; Val val = Rapids.exec(tree); assertEquals(val.getNum(), 6.0, 1e-5); }
@Test public void changeKeyFrameTest() { Frame res = null; try { fr = new TestFrameBuilder() .withName("testFrame") .withColNames("ColA") .withVecTypes(Vec.T_NUM) .withDataForCol(0, ard(1, 2)) .build(); String tree = "( append testFrame 42 'appended' )"; Val val = Rapids.exec(tree); res = val.getFrame(); res._key = fr._key; DKV.put(fr._key, res); } finally { res.delete(); } }
private String[] getTEColumns(Env env, Env.StackHelp stk, AstRoot asts[]) { if (asts[2] instanceof AstStrList) { AstStrList teColumns = ((AstStrList) asts[2]); return teColumns._strs; } else if( asts[2] instanceof AstStr) { String teColumn = stk.track(asts[2].exec(env)).getStr(); return new String[]{teColumn}; } else throw new IllegalStateException("Couldn't parse `teColumns` parameter"); }
private byte getDataLeakageHandlingStrategy(Env env, Env.StackHelp stk, AstRoot asts[]) { String strategy = stk.track(asts[5].exec(env)).getStr(); if(strategy.equals("kfold")){ return TargetEncoder.DataLeakageHandlingStrategy.KFold; } else if (strategy.equals("loo")) { return TargetEncoder.DataLeakageHandlingStrategy.LeaveOneOut; } else if(strategy.equals("loo")) { return TargetEncoder.DataLeakageHandlingStrategy.None; } else { return TargetEncoder.DataLeakageHandlingStrategy.None; } }