I am using H2O in python to make a Generalized Linear Model, binary classification problem, I made the model using
glm_fit_lambda_search = H2OGeneralizedLinearEstimator( family='binomial',
model_id='glm_fit_lambda_search',
lambda_search=True )
glm_fit_lambda_search.train( x = x,
y = y,
training_frame = trainH2O,
validation_frame = testH2O )
Now I want to plot the ROC curve of the model, how can I do that?
Also I want to plot multiple ROC curves for comparison
Here is the question in R, How to directly plot ROC of h2o model object in R, How can I do this in python?
try this:
performace = glm_fit_lambda_search.model_performance(train=True)
performace.plot()
should work in theory, I'm not able to verify right now. This will plot the performance on the "train" set.