unknown family: negativebinomial is not recognized when used with H2OGridSearch

Description

This is not working with lambda_search=True

Example:

import h2o
from h2o.estimators.glm import H2OGeneralizedLinearEstimator
from h2o.grid.grid_search import H2OGridSearch

prostate = h2o.import_file("/Users/wendycwong/h2o-3/smalldata/prostate/prostate_complete.csv.zip"))
myY = "GLEASON"
myX = ["ID","AGE","RACE","CAPSULE","DCAPS","PSA","VOL","DPROS"]
hyper_parameters =

{'alpha': [0, 0.5, 0.99], 'theta' : [0.000000001, 0.01, 0.1, 0.5, 1]}
model_h2o_grid_search = H2OGridSearch(H2OGeneralizedLinearEstimator(family="negativebinomial", lambda_search=True),hyper_parameters)
model_h2o_grid_search.train(x=myX, y=myY, training_frame=prostate)
print(model_h2o_grid_search.get_grid("residual_deviance"))
assert(len(model_h2o_grid_search.get_grid())==15)

Here is the failure stack:

failure_details: unknown family negativebinomial
failure_stack_traces: java.lang.RuntimeException: unknown family negativebinomial
at hex.glm.ComputationState.deviance(ComputationState.java:257)
at hex.glm.GLM$GLMDriver.computeSubmodel(GLM.java:1184)
at hex.glm.GLM$GLMDriver.computeImpl(GLM.java:1266)
at hex.ModelBuilder$Driver.compute2(ModelBuilder.java:218)
at hex.glm.GLM$GLMDriver.compute2(GLM.java:581)
at water.H2O$H2OCountedCompleter.compute(H2O.java:1386)
at jsr166y.CountedCompleter.exec(CountedCompleter.java:468)
at jsr166y.ForkJoinTask.doExec(ForkJoinTask.java:263)
at jsr166y.ForkJoinPool$WorkQueue.runTask(ForkJoinPool.java:974)
at jsr166y.ForkJoinPool.runWorker(ForkJoinPool.java:1477)
at jsr166y.ForkJoinWorkerThread.run(ForkJoinWorkerThread.java:104)

Assignee

Wendy

Fix versions

Reporter

jean charles

Support ticket URL

None

Labels

None

Affected Spark version

None

Customer Request Type

None

Task progress

None

ReleaseNotesHidden

None

CustomerVisible

No

Priority

Major
Configure