예제 #1
0
파일: GLMModel.java 프로젝트: vsynych/h2o-3
 public GLMOutput(GLM glm) {
   super(glm);
   _dinfo = glm._dinfo;
   if (!glm.hasWeightCol()) {
     _dinfo = (DataInfo) _dinfo.clone();
     _dinfo._adaptedFrame =
         new Frame(_dinfo._adaptedFrame.names().clone(), _dinfo._adaptedFrame.vecs().clone());
     _dinfo.dropWeights();
   }
   _scoringDinfo = _dinfo.scoringInfo();
   String[] cnames = glm._dinfo.coefNames();
   String[] names = _dinfo._adaptedFrame._names;
   String[][] domains = _dinfo._adaptedFrame.domains();
   int id = glm._generatedWeights == null ? -1 : ArrayUtils.find(names, glm._generatedWeights);
   if (id >= 0) {
     String[] ns = new String[names.length - 1];
     String[][] ds = new String[domains.length - 1][];
     System.arraycopy(names, 0, ns, 0, id);
     System.arraycopy(domains, 0, ds, 0, id);
     System.arraycopy(names, id + 1, ns, id, ns.length - id);
     System.arraycopy(domains, id + 1, ds, id, ds.length - id);
     names = ns;
     domains = ds;
   }
   _names = names;
   _domains = domains;
   _coefficient_names = Arrays.copyOf(cnames, cnames.length + 1);
   _coefficient_names[_coefficient_names.length - 1] = "Intercept";
   _binomial = glm._parms._family == Family.binomial;
   _nclasses = glm.nclasses();
   _multinomial = _nclasses > 2;
 }
예제 #2
0
파일: DataInfo.java 프로젝트: liaochy/h2o-3
 public DataInfo validDinfo(Frame valid) {
   DataInfo res =
       new DataInfo(
           _adaptedFrame,
           null,
           1,
           _useAllFactorLevels,
           TransformType.NONE,
           TransformType.NONE,
           _skipMissing,
           _imputeMissing,
           false,
           _weights,
           _offset,
           _fold);
   res._adaptedFrame = new Frame(_adaptedFrame.names(), valid.vecs(_adaptedFrame.names()));
   res._valid = true;
   return res;
 }
예제 #3
0
 public DataInfo scoringInfo() {
   DataInfo res =
       new DataInfo(
           _adaptedFrame,
           null,
           1,
           _useAllFactorLevels,
           TransformType.NONE,
           TransformType.NONE,
           _skipMissing,
           _imputeMissing,
           !_skipMissing,
           _weights,
           _offset,
           _fold);
   res._adaptedFrame = null;
   res._weights = false;
   res._offset = false;
   res._fold = false;
   res._responses = 0;
   res._valid = true;
   res._interactions = _interactions;
   return res;
 }