Skip to content

Commit b2c4762

Browse files
committed
Implement tweedie and gaussian tests
1 parent 2291297 commit b2c4762

2 files changed

Lines changed: 311 additions & 46 deletions

File tree

h2o-algos/src/main/java/hex/glm/GLM.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3542,10 +3542,10 @@ private void scorePostProcessingRestrictedModelROEnabled(Frame train, long t1) {
35423542
if ((mtrain != null) && (_valid != null)) {
35433543
_scoringHistoryRemoveOffsetEnabled.addIterationScore(true, true, _state._iter, _state.likelihood(),
35443544
_state.objective(), _state.deviance(), ((GLMMetrics) _model._output._validation_metrics_restricted_model_ro).residual_deviance(),
3545-
mtrain._nobs, _model._output._validation_metrics_restricted_model_ro._nobs, _state.lambda(), _state.alpha());
3545+
_model._output._training_metrics_restricted_model_ro._nobs, _model._output._validation_metrics_restricted_model_ro._nobs, _state.lambda(), _state.alpha());
35463546
} else { // only doing training deviance
35473547
_scoringHistoryRemoveOffsetEnabled.addIterationScore(true, false, _state._iter, _state.likelihood(),
3548-
_state.objective(), _state.deviance(), Double.NaN, mtrain._nobs, 1, _state.lambda(),
3548+
_state.objective(), _state.deviance(), Double.NaN, _model._output._training_metrics_restricted_model_ro._nobs, 1, _state.lambda(),
35493549
_state.alpha());
35503550
}
35513551
}

0 commit comments

Comments
 (0)