private Frame execRapidsAndGetFrame(String astTree) { Val val = Rapids.exec(astTree); return register(val.getFrame()); }
private double getSmoothing(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[10].exec(env)).getNum(); }
@Override public ValFrame apply(Env env, Env.StackHelp stk, AstRoot asts[]) { Word2VecModel model = (Word2VecModel) stk.track(asts[1].exec(env)).getModel(); return new ValFrame(model.toFrame()); }
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; } }
private String getFoldColumnName(Env env, Env.StackHelp stk, AstRoot asts[]) { try { String str = stk.track(asts[7].exec(env)).getStr(); if(str.equals("")) return null; return str; } catch (IllegalArgumentException ex) { return null; } }
private Frame getFrameToTransform(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[3].exec(env)).getFrame(); }
private double getSeed(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[12].exec(env)).getNum(); } }
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"); }
@Override public ValFrame apply(Env env, Env.StackHelp stk, AstRoot asts[]) { Word2VecModel model = (Word2VecModel) stk.track(asts[1].exec(env)).getModel(); return new ValFrame(model.toFrame()); }
private Frame getTrainingFrame(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[1].exec(env)).getFrame(); }
private boolean getWithBlending(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[8].exec(env)).getNum() == 1; }
private String getFoldColumnName(Env env, Env.StackHelp stk, AstRoot asts[]) { try { String str = stk.track(asts[4].exec(env)).getStr(); if(str.equals("")) return null; return str; } catch (IllegalArgumentException ex) { return null; } }
public long rowsWithNa() { if( _rowsWithNa!=-1 ) return _rowsWithNa; String x = String.format("(na.omit %s)", _fr._key); Val res = Rapids.exec(x); Frame f = res.getFrame(); long cnt = _fr.numRows() - f.numRows(); f.delete(); return (_rowsWithNa=cnt); }
private boolean getWithImputation(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[5].exec(env)).getNum() == 1; }
private String getTargetColumnName(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[3].exec(env)).getStr(); }
_train = Rapids.exec(String.format("(na.omit %s)", tranRebalanced._key)).getFrame(); // remove NA rows DKV.remove(tranRebalanced._key); checkMemoryFootPrint();
private double getNoise(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[11].exec(env)).getNum(); }
private String getTargetColumnName(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[6].exec(env)).getStr(); }
@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 double getInflectionPoint(Env env, Env.StackHelp stk, AstRoot asts[]) { return stk.track(asts[9].exec(env)).getNum(); }