Ejemplo n.º 1
0
def build_and_test(a_node, pp, datasets, algos, algo_additional_default_params):
    ####################################################################################################
    # Build and do basic validation checks on models
    ####################################################################################################
    models_to_build = [
        ModelSpec.for_dataset('kmeans_prostate', 'kmeans', datasets['prostate_clustering'], { 'k': 2 } ),
    
        ModelSpec.for_dataset('glm_prostate_regression', 'glm', datasets['prostate_regression'], {'family': 'gaussian'} ),
    
        ModelSpec.for_dataset('glm_prostate_binomial', 'glm', datasets['prostate_binomial'], {'family': 'binomial'} ),
        ModelSpec.for_dataset('glm_airlines_binomial', 'glm', datasets['airlines_binomial'], {'response_column': 'IsDepDelayed', 'family': 'binomial' } ),
        ModelSpec.for_dataset('glm_iris_multinomial', 'glm', datasets['iris_multinomial'], {'response_column': 'class', 'family': 'multinomial' } ),
    
        ModelSpec.for_dataset('deeplearning_prostate_regression', 'deeplearning', datasets['prostate_regression'], { 'epochs': 1, 'loss': 'Quadratic' } ),
        ModelSpec.for_dataset('deeplearning_prostate_binomial', 'deeplearning', datasets['prostate_binomial'], { 'epochs': 1, 'hidden': [20, 20], 'loss': 'CrossEntropy' } ),
        ModelSpec.for_dataset('deeplearning_airlines_binomial', 'deeplearning', datasets['airlines_binomial'], { 'epochs': 1, 'hidden': [10, 10], 'loss': 'CrossEntropy' } ),
        ModelSpec.for_dataset('deeplearning_iris_multinomial', 'deeplearning', datasets['iris_multinomial'], { 'epochs': 1, 'loss': 'CrossEntropy' } ),
    
        ModelSpec.for_dataset('gbm_prostate_regression', 'gbm', datasets['prostate_regression'], { 'ntrees': 5, 'distribution': 'gaussian' } ),
        ModelSpec.for_dataset('gbm_prostate_binomial', 'gbm', datasets['prostate_binomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
        ModelSpec.for_dataset('gbm_airlines_binomial', 'gbm', datasets['airlines_binomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
        ModelSpec.for_dataset('gbm_iris_multinomial', 'gbm', datasets['iris_multinomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
       ]
    
    # For grid testing, don't build any non-grid models:
    # models_to_build = []

    built_models = {}
    for model_spec in models_to_build:
        model = model_spec.build_and_validate_model(a_node)
        built_models[model_spec['dest_key']] = model
    
    

    grids_to_build = [
        GridSpec.for_dataset('kmeans_prostate_grid', 'kmeans', datasets['prostate_clustering'], { }, { 'k': [2, 3, 4] } ),
    
        GridSpec.for_dataset('glm_prostate_regression_grid', 'glm', datasets['prostate_regression'], {'family': 'gaussian'}, { 'lambda': [0.0001, 0.001, 0.01, 0.1] } ),
    
        GridSpec.for_dataset('glm_prostate_binomial_grid', 'glm', datasets['prostate_binomial'], {'family': 'binomial'}, { 'lambda': [0.0001, 0.001, 0.01, 0.1] }  ),
        GridSpec.for_dataset('glm_airlines_binomial_grid', 'glm', datasets['airlines_binomial'], {'response_column': 'IsDepDelayed', 'family': 'binomial'}, { 'lambda': [0.0001, 0.001, 0.01, 0.025] } ),
        GridSpec.for_dataset('glm_iris_multinomial_grid', 'glm', datasets['iris_multinomial'], {'response_column': 'class', 'family': 'multinomial'}, { 'lambda': [0.0001, 0.001, 0.01, 0.025] } ),
    
        GridSpec.for_dataset('deeplearning_prostate_regression_grid', 'deeplearning', datasets['prostate_regression'], { 'loss': 'Quadratic' }, { 'epochs': [0.1, 0.5, 1] } ),
        GridSpec.for_dataset('deeplearning_prostate_binomial_grid', 'deeplearning', datasets['prostate_binomial'], { 'hidden': [20, 20], 'loss': 'CrossEntropy' }, { 'epochs': [0.1, 0.5, 1] }  ),
        GridSpec.for_dataset('deeplearning_airlines_binomial_grid', 'deeplearning', datasets['airlines_binomial'], { 'hidden': [10, 10], 'loss': 'CrossEntropy' }, { 'epochs': [0.1, 0.5, 1] }  ),
        GridSpec.for_dataset('deeplearning_iris_multinomial_grid', 'deeplearning', datasets['iris_multinomial'], { 'loss': 'CrossEntropy' }, { 'epochs': [0.1, 0.5, 1] }  ),
    
        GridSpec.for_dataset('gbm_prostate_regression_grid', 'gbm', datasets['prostate_regression'], { 'max_depth': 3 }, { 'ntrees': [1, 5, 10], 'distribution': ["gaussian", "poisson", "gamma", "tweedie"] }  ),
        GridSpec.for_dataset('gbm_prostate_binomial_grid', 'gbm', datasets['prostate_binomial'], {  }, { 'ntrees': [5, 7], 'max_depth': [1, 3, 5] } ),
        GridSpec.for_dataset('gbm_airlines_binomial_grid', 'gbm', datasets['airlines_binomial'], { 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),
        GridSpec.for_dataset('gbm_iris_multinomial_grid', 'gbm', datasets['iris_multinomial'], { 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),
        # TODO: this should trigger a parameter validation error, but instead the non-grid ntrees silently overrides the grid values:        GridSpec.for_dataset('gbm_iris_multinomial_grid', 'gbm', datasets['iris_multinomial'], { 'ntrees': 5, 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),

        # Test stopping criteria:
        GridSpec.for_dataset('gbm_prostate_regression_grid_max_3', 'gbm', datasets['prostate_regression'], { 'max_depth': 3 }, { 'ntrees': [1, 2, 4], 'distribution': ["gaussian", "poisson", "gamma", "tweedie"] }, { 'strategy': "RandomDiscrete", 'max_models': 3 } ),
        GridSpec.for_dataset('gbm_prostate_regression_grid_max_20mS', 'gbm', datasets['prostate_regression'], { 'max_depth': 3 }, { 'ntrees': [1, 2, 4], 'distribution': ["gaussian", "poisson", "gamma", "tweedie"] }, { 'strategy': "RandomDiscrete", 'max_runtime_secs': 0.020 } ),
       ]
    
    for grid_spec in grids_to_build:
        grid = grid_spec.build_and_validate_grid(a_node)

        for model_key in grid['model_ids']:
            model_key = model_key['name']
            built_models[model_key] = a_node.models(key=model_key)

    # test search limits: max_models
    grid = a_node.grid(key='gbm_prostate_regression_grid_max_3')
    assert len(grid['model_ids']) == 3, "FAIL: using max_models, expected a max of 3 models, got: " + str(len(grid['model_ids']))

    # test search limits: max_runtime_secs
    grid = a_node.grid(key='gbm_prostate_regression_grid_max_20mS')
    assert len(grid['model_ids']) < 12, "FAIL: using max_runtime_secs, expected less than 12 models, got: " + str(len(grid['model_ids']))

#    grid = a_node.grid(key='kmeans_prostate_grid', sort_by='', decreasing=True)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='totss', decreasing=True)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='tot_withinss', decreasing=True)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='betweenss', decreasing=True)

    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='totss', decreasing=False)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='tot_withinss', decreasing=False)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='betweenss', decreasing=False)

#    import sys
#    sys.exit(0)
    #######################################
    # Test default parameters validation for each model builder
    #
    if h2o_test_utils.isVerbose(): print 'Testing ModelBuilder default parameters. . .'
    model_builders = a_node.model_builders(timeoutSecs=240)['model_builders']
    
    # Do we know about all of them?
    server_algos = model_builders.keys()
    assert len(set(server_algos) - set(algos)) == 0, "FAIL: Our set of algos doesn't match what the server knows about.  Ours: " + repr(algos) + "; server's: " + repr(server_algos)
    
    for algo, model_builder in model_builders.iteritems():
        parameters_list = model_builder['parameters']
        test_parameters = { value['name'] : value['default_value'] for value in parameters_list } # collect default parameters
        if algo in algo_additional_default_params:
            test_parameters.update(algo_additional_default_params[algo])
    
        if h2o_test_utils.isVerboser(): print 'Testing ' + algo + ' with params: ' + repr(test_parameters)
    
        parameters_validation = a_node.validate_model_parameters(algo=algo, training_frame=None, parameters=test_parameters, timeoutSecs=240) # synchronous
        assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
        h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    
        expected_count = 0
        if expected_count != parameters_validation['error_count']:
            print "validation errors: "
            pp.pprint(parameters_validation)
        assert expected_count == parameters_validation['error_count'], "FAIL: " + str(expected_count) + " != error_count in good-parameters parameters validation result."
    
    
    #######################################
    # Test DeepLearning parameters validation
    #
    # Default parameters:
    if h2o_test_utils.isVerbose(): print 'Testing DeepLearning default parameters. . .'
    model_builder = a_node.model_builders(algo='deeplearning', timeoutSecs=240)['model_builders']['deeplearning']
    dl_test_parameters_list = model_builder['parameters']
    dl_test_parameters = {value['name'] : value['default_value'] for value in dl_test_parameters_list}
    
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print "validation errors: "
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation['error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."
    
    # Good parameters (note: testing with null training_frame):
    if h2o_test_utils.isVerbose(): print 'Testing DeepLearning good parameters. . .'
    dl_test_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]" }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print "validation errors: "
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation['error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."
    
    # Bad parameters (hidden is null):
    # (note: testing with null training_frame)
    if h2o_test_utils.isVerbose(): print 'Testing DeepLearning bad parameters, null training_frame. . .'
    dl_test_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]", 'input_dropout_ratio': 27 }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (input_dropout_ratio)."
    h2o.H2O.verboseprint("Good params validation messages: ", repr(parameters_validation))
    assert 0 != parameters_validation['error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(parameters_validation)
    
    found_expected_error = False
    for validation_message in parameters_validation['messages']:
        if validation_message['message_type'] == 'ERRR' and validation_message['field_name'] == 'input_dropout_ratio':
            found_expected_error = True
    assert found_expected_error, "FAIL: Failed to find error message about input_dropout_ratio in the validation messages."
    
    # Bad parameters (no response_column):
    if h2o_test_utils.isVerbose(): print 'Testing DeepLearning bad parameters, null response_column. . .'
    dl_test_parameters = {'hidden': "[10, 20, 10]" }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame='prostate_binomial', parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (response_column)."
    h2o.H2O.verboseprint("Good params validation messages: ", repr(parameters_validation))
    assert 0 != parameters_validation['error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(parameters_validation)
    
    
    #######################################
    # Try to build DeepLearning model for Prostate but with bad parameters; we should get a ModelParametersSchema with the error.
    if h2o_test_utils.isVerbose(): print 'About to try to build a DeepLearning model with bad parameters. . .'
    dl_prostate_bad_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]", 'input_dropout_ratio': 27  }
    parameters_validation = a_node.build_model(algo='deeplearning', model_id='deeplearning_prostate_binomial_bad', training_frame='prostate_binomial', parameters=dl_prostate_bad_parameters, timeoutSecs=240) # synchronous
    h2o_test_utils.validate_validation_messages(parameters_validation, ['input_dropout_ratio'])
    assert parameters_validation['__http_response']['status_code'] == requests.codes.precondition_failed, "FAIL: expected 412 Precondition Failed from a bad build request, got: " + str(parameters_validation['__http_response']['status_code'])
    if h2o_test_utils.isVerbose(): print 'Done trying to build DeepLearning model with bad parameters.'
    
    #####################################
    # Early test of predict()
    # TODO: remove after we remove the early exit
    p = a_node.predict(model='deeplearning_airlines_binomial', frame='airlines_binomial', predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_airlines_binomial', 'airlines_binomial', 43978, predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_frame_exists(a_node, 'deeplearning_airlines_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_airlines_binomial', " on: ", 'airlines_binomial', ":  ", repr(p))
Ejemplo n.º 2
0
def build_and_test(a_node, pp, datasets, algos, algo_additional_default_params):
    ####################################################################################################
    # Build and do basic validation checks on models
    ####################################################################################################
    models_to_build = [
        ModelSpec.for_dataset('kmeans_prostate', 'kmeans', datasets['prostate_clustering'], { 'k': 2 } ),
    
        ModelSpec.for_dataset('glm_prostate_regression', 'glm', datasets['prostate_regression'], {'family': 'gaussian'} ),
    
        ModelSpec.for_dataset('glm_prostate_binomial', 'glm', datasets['prostate_binomial'], {'family': 'binomial'} ),
        # TODO: Crashes: ModelSpec('glm_airlines_binomial', 'glm', 'airlines_binomial', {'response_column': 'IsDepDelayed', 'do_classification': True, 'family': 'binomial'}, 'Binomial'),
        # Multinomial doesn't make sense for glm: ModelSpec('glm_iris_multinomial', 'glm', iris_multinomial, {'response_column': 'class', 'do_classification': True, 'family': 'gaussian'}, 'Regression'),
    
        ModelSpec.for_dataset('deeplearning_prostate_regression', 'deeplearning', datasets['prostate_regression'], { 'epochs': 1, 'loss': 'Automatic' } ),
        # TODO: add toEnum of the response column and put back:    ModelSpec.for_dataset('deeplearning_prostate_binomial', 'deeplearning', datasets['prostate_binomial'], { 'epochs': 1, 'hidden': [20, 20], 'loss': 'CrossEntropy' } ),
        ModelSpec.for_dataset('deeplearning_airlines_binomial', 'deeplearning', datasets['airlines_binomial'], { 'epochs': 1, 'hidden': [10, 10], 'loss': 'CrossEntropy' } ),
        ModelSpec.for_dataset('deeplearning_iris_multinomial', 'deeplearning', datasets['iris_multinomial'], { 'epochs': 1, 'loss': 'CrossEntropy' } ),
    
        ModelSpec.for_dataset('gbm_prostate_regression', 'gbm', datasets['prostate_regression'], { 'ntrees': 5, 'distribution': 'gaussian' } ),
        # TODO: add toEnum of the response column and put back:        ModelSpec.for_dataset('gbm_prostate_binomial', 'gbm', datasets['prostate_binomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
        ModelSpec.for_dataset('gbm_airlines_binomial', 'gbm', datasets['airlines_binomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
        ModelSpec.for_dataset('gbm_iris_multinomial', 'gbm', datasets['iris_multinomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
       ]
    
    built_models = {}
    for model_spec in models_to_build:
        model = model_spec.build_and_validate_model(a_node)
        built_models[model_spec['dest_key']] = model
    
    
    #######################################
    # Test default parameters validation for each model builder
    #
    if h2o_test_utils.isVerbose(): print 'Testing ModelBuilder default parameters. . .'
    model_builders = a_node.model_builders(timeoutSecs=240)['model_builders']
    
    # Do we know about all of them?
    server_algos = model_builders.keys()
    assert len(set(server_algos) - set(algos)) == 0, "FAIL: Our set of algos doesn't match what the server knows about.  Ours: " + repr(algos) + "; server's: " + repr(server_algos)
    
    for algo, model_builder in model_builders.iteritems():
        parameters_list = model_builder['parameters']
        test_parameters = { value['name'] : value['default_value'] for value in parameters_list } # collect default parameters
        if algo in algo_additional_default_params:
            test_parameters.update(algo_additional_default_params[algo])
    
        parameters_validation = a_node.validate_model_parameters(algo=algo, training_frame=None, parameters=test_parameters, timeoutSecs=240) # synchronous
        assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
        h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    
        expected_count = 0
        if expected_count != parameters_validation['error_count']:
            print "validation errors: "
            pp.pprint(parameters_validation)
        assert expected_count == parameters_validation['error_count'], "FAIL: " + str(expected_count) + " != error_count in good-parameters parameters validation result."
    
    
    #######################################
    # Test DeepLearning parameters validation
    #
    # Default parameters:
    model_builder = a_node.model_builders(algo='deeplearning', timeoutSecs=240)['model_builders']['deeplearning']
    dl_test_parameters_list = model_builder['parameters']
    dl_test_parameters = {value['name'] : value['default_value'] for value in dl_test_parameters_list}
    
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print "validation errors: "
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation['error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."
    
    # Good parameters (note: testing with null training_frame):
    dl_test_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]" }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print "validation errors: "
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation['error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."
    
    # Bad parameters (hidden is null):
    # (note: testing with null training_frame)
    dl_test_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]", 'input_dropout_ratio': 27 }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (input_dropout_ratio)."
    h2o.H2O.verboseprint("Good params validation messages: ", repr(parameters_validation))
    assert 0 != parameters_validation['error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(parameters_validation)
    
    found_expected_error = False
    for validation_message in parameters_validation['messages']:
        if validation_message['message_type'] == 'ERROR' and validation_message['field_name'] == 'input_dropout_ratio':
            found_expected_error = True
    assert found_expected_error, "FAIL: Failed to find error message about input_dropout_ratio in the validation messages."
    
    # Bad parameters (no response_column):
    dl_test_parameters = {'hidden': "[10, 20, 10]" }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame='prostate_binomial', parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (response_column)."
    h2o.H2O.verboseprint("Good params validation messages: ", repr(parameters_validation))
    assert 0 != parameters_validation['error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(parameters_validation)
    
    
    #######################################
    # Try to build DeepLearning model for Prostate but with bad parameters; we should get a ModelParametersSchema with the error.
    if h2o_test_utils.isVerbose(): print 'About to try to build a DeepLearning model with bad parameters. . .'
    dl_prostate_bad_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]", 'input_dropout_ratio': 27  }
    parameters_validation = a_node.build_model(algo='deeplearning', model_id='deeplearning_prostate_binomial_bad', training_frame='prostate_binomial', parameters=dl_prostate_bad_parameters, timeoutSecs=240) # synchronous
    h2o_test_utils.validate_validation_messages(parameters_validation, ['input_dropout_ratio'])
    assert parameters_validation['__http_response']['status_code'] == requests.codes.precondition_failed, "FAIL: expected 412 Precondition Failed from a bad build request, got: " + str(parameters_validation['__http_response']['status_code'])
    if h2o_test_utils.isVerbose(): print 'Done trying to build DeepLearning model with bad parameters.'
    
    #####################################
    # Early test of predict()
    # TODO: remove after we remove the early exit
    p = a_node.predict(model='deeplearning_airlines_binomial', frame='airlines_binomial', predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_airlines_binomial', 'airlines_binomial', 43978, predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_frame_exists(a_node, 'deeplearning_airlines_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_airlines_binomial', " on: ", 'airlines_binomial', ":  ", repr(p))
Ejemplo n.º 3
0
def build_and_test(a_node, pp, datasets, algos, algo_additional_default_params):
    ####################################################################################################
    # Build and do basic validation checks on models
    ####################################################################################################
    models_to_build = [
        ModelSpec.for_dataset('kmeans_prostate', 'kmeans', datasets['prostate_clustering'], { 'k': 2 } ),
    
        ModelSpec.for_dataset('glm_prostate_regression', 'glm', datasets['prostate_regression'], {'family': 'gaussian'} ),
    
        ModelSpec.for_dataset('glm_prostate_binomial', 'glm', datasets['prostate_binomial'], {'family': 'binomial'} ),
        ModelSpec.for_dataset('glm_airlines_binomial', 'glm', datasets['airlines_binomial'], {'response_column': 'IsDepDelayed', 'family': 'binomial' } ),
        ModelSpec.for_dataset('glm_iris_multinomial', 'glm', datasets['iris_multinomial'], {'response_column': 'class', 'family': 'multinomial' } ),
    
        ModelSpec.for_dataset('deeplearning_prostate_regression', 'deeplearning', datasets['prostate_regression'], { 'epochs': 1, 'loss': 'Quadratic' } ),
        ModelSpec.for_dataset('deeplearning_prostate_binomial', 'deeplearning', datasets['prostate_binomial'], { 'epochs': 1, 'hidden': [20, 20], 'loss': 'CrossEntropy' } ),
        ModelSpec.for_dataset('deeplearning_airlines_binomial', 'deeplearning', datasets['airlines_binomial'], { 'epochs': 1, 'hidden': [10, 10], 'loss': 'CrossEntropy' } ),
        ModelSpec.for_dataset('deeplearning_iris_multinomial', 'deeplearning', datasets['iris_multinomial'], { 'epochs': 1, 'loss': 'CrossEntropy' } ),
    
        ModelSpec.for_dataset('gbm_prostate_regression', 'gbm', datasets['prostate_regression'], { 'ntrees': 5, 'distribution': 'gaussian' } ),
        ModelSpec.for_dataset('gbm_prostate_binomial', 'gbm', datasets['prostate_binomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
        ModelSpec.for_dataset('gbm_airlines_binomial', 'gbm', datasets['airlines_binomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
        ModelSpec.for_dataset('gbm_iris_multinomial', 'gbm', datasets['iris_multinomial'], { 'ntrees': 5, 'distribution': 'multinomial' } ),
       ]
    
    built_models = {}
    for model_spec in models_to_build:
        model = model_spec.build_and_validate_model(a_node)
        built_models[model_spec['dest_key']] = model
    
    

    grids_to_build = [
        GridSpec.for_dataset('kmeans_prostate_grid', 'kmeans', datasets['prostate_clustering'], { }, { 'k': [2, 3, 4] } ),
    
        GridSpec.for_dataset('glm_prostate_regression_grid', 'glm', datasets['prostate_regression'], {'family': 'gaussian'}, { 'lambda': [0.0001, 0.001, 0.01, 0.1] } ),
    
        GridSpec.for_dataset('glm_prostate_binomial_grid', 'glm', datasets['prostate_binomial'], {'family': 'binomial'}, { 'lambda': [0.0001, 0.001, 0.01, 0.1] }  ),
        GridSpec.for_dataset('glm_airlines_binomial_grid', 'glm', datasets['airlines_binomial'], {'response_column': 'IsDepDelayed', 'family': 'binomial'}, { 'lambda': [0.0001, 0.001, 0.01, 0.025] } ),
        GridSpec.for_dataset('glm_iris_multinomial_grid', 'glm', datasets['iris_multinomial'], {'response_column': 'class', 'family': 'multinomial'}, { 'lambda': [0.0001, 0.001, 0.01, 0.025] } ),
    
        GridSpec.for_dataset('deeplearning_prostate_regression_grid', 'deeplearning', datasets['prostate_regression'], { 'loss': 'Quadratic' }, { 'epochs': [0.1, 0.5, 1] } ),
        GridSpec.for_dataset('deeplearning_prostate_binomial_grid', 'deeplearning', datasets['prostate_binomial'], { 'hidden': [20, 20], 'loss': 'CrossEntropy' }, { 'epochs': [0.1, 0.5, 1] }  ),
        GridSpec.for_dataset('deeplearning_airlines_binomial_grid', 'deeplearning', datasets['airlines_binomial'], { 'hidden': [10, 10], 'loss': 'CrossEntropy' }, { 'epochs': [0.1, 0.5, 1] }  ),
        GridSpec.for_dataset('deeplearning_iris_multinomial_grid', 'deeplearning', datasets['iris_multinomial'], { 'loss': 'CrossEntropy' }, { 'epochs': [0.1, 0.5, 1] }  ),
    
        GridSpec.for_dataset('gbm_prostate_regression_grid', 'gbm', datasets['prostate_regression'], { 'max_depth': 3 }, { 'ntrees': [1, 5, 10], 'distribution': ["gaussian", "poisson", "gamma", "tweedie"] }  ),
        GridSpec.for_dataset('gbm_prostate_binomial_grid', 'gbm', datasets['prostate_binomial'], {  }, { 'ntrees': [5, 7], 'max_depth': [1, 3, 5] } ),
        GridSpec.for_dataset('gbm_airlines_binomial_grid', 'gbm', datasets['airlines_binomial'], { 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),
        GridSpec.for_dataset('gbm_iris_multinomial_grid', 'gbm', datasets['iris_multinomial'], { 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),
        # TODO: this should trigger a parameter validation error, but instead the non-grid ntrees silently overrides the drid values:        GridSpec.for_dataset('gbm_iris_multinomial_grid', 'gbm', datasets['iris_multinomial'], { 'ntrees': 5, 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),
       ]
    
    for grid_spec in grids_to_build:
        grid = grid_spec.build_and_validate_grid(a_node)

        for model_key in grid['model_ids']:
            model_key = model_key['name']
            built_models[model_key] = a_node.models(key=model_key)

#    grid = a_node.grid(key='kmeans_prostate_grid', sort_by='', sort_order='desc')
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='totss', sort_order='desc')
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='tot_withinss', sort_order='desc')
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='betweenss', sort_order='desc')

    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='totss', sort_order='asc')
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='tot_withinss', sort_order='asc')
    h2o_test_utils.fetch_and_validate_grid_sort(a_node, key='kmeans_prostate_grid', sort_by='betweenss', sort_order='asc')


    #######################################
    # Test default parameters validation for each model builder
    #
    if h2o_test_utils.isVerbose(): print 'Testing ModelBuilder default parameters. . .'
    model_builders = a_node.model_builders(timeoutSecs=240)['model_builders']
    
    # Do we know about all of them?
    server_algos = model_builders.keys()
    assert len(set(server_algos) - set(algos)) == 0, "FAIL: Our set of algos doesn't match what the server knows about.  Ours: " + repr(algos) + "; server's: " + repr(server_algos)
    
    for algo, model_builder in model_builders.iteritems():
        parameters_list = model_builder['parameters']
        test_parameters = { value['name'] : value['default_value'] for value in parameters_list } # collect default parameters
        if algo in algo_additional_default_params:
            test_parameters.update(algo_additional_default_params[algo])
    
        parameters_validation = a_node.validate_model_parameters(algo=algo, training_frame=None, parameters=test_parameters, timeoutSecs=240) # synchronous
        assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
        h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    
        expected_count = 0
        if expected_count != parameters_validation['error_count']:
            print "validation errors: "
            pp.pprint(parameters_validation)
        assert expected_count == parameters_validation['error_count'], "FAIL: " + str(expected_count) + " != error_count in good-parameters parameters validation result."
    
    
    #######################################
    # Test DeepLearning parameters validation
    #
    # Default parameters:
    model_builder = a_node.model_builders(algo='deeplearning', timeoutSecs=240)['model_builders']['deeplearning']
    dl_test_parameters_list = model_builder['parameters']
    dl_test_parameters = {value['name'] : value['default_value'] for value in dl_test_parameters_list}
    
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print "validation errors: "
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation['error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."
    
    # Good parameters (note: testing with null training_frame):
    dl_test_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]" }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ", repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print "validation errors: "
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation['error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."
    
    # Bad parameters (hidden is null):
    # (note: testing with null training_frame)
    dl_test_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]", 'input_dropout_ratio': 27 }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame=None, parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (input_dropout_ratio)."
    h2o.H2O.verboseprint("Good params validation messages: ", repr(parameters_validation))
    assert 0 != parameters_validation['error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(parameters_validation)
    
    found_expected_error = False
    for validation_message in parameters_validation['messages']:
        if validation_message['message_type'] == 'ERRR' and validation_message['field_name'] == 'input_dropout_ratio':
            found_expected_error = True
    assert found_expected_error, "FAIL: Failed to find error message about input_dropout_ratio in the validation messages."
    
    # Bad parameters (no response_column):
    dl_test_parameters = {'hidden': "[10, 20, 10]" }
    parameters_validation = a_node.validate_model_parameters(algo='deeplearning', training_frame='prostate_binomial', parameters=dl_test_parameters, timeoutSecs=240) # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (response_column)."
    h2o.H2O.verboseprint("Good params validation messages: ", repr(parameters_validation))
    assert 0 != parameters_validation['error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(parameters_validation)
    
    
    #######################################
    # Try to build DeepLearning model for Prostate but with bad parameters; we should get a ModelParametersSchema with the error.
    if h2o_test_utils.isVerbose(): print 'About to try to build a DeepLearning model with bad parameters. . .'
    dl_prostate_bad_parameters = {'response_column': 'CAPSULE', 'hidden': "[10, 20, 10]", 'input_dropout_ratio': 27  }
    parameters_validation = a_node.build_model(algo='deeplearning', model_id='deeplearning_prostate_binomial_bad', training_frame='prostate_binomial', parameters=dl_prostate_bad_parameters, timeoutSecs=240) # synchronous
    h2o_test_utils.validate_validation_messages(parameters_validation, ['input_dropout_ratio'])
    assert parameters_validation['__http_response']['status_code'] == requests.codes.precondition_failed, "FAIL: expected 412 Precondition Failed from a bad build request, got: " + str(parameters_validation['__http_response']['status_code'])
    if h2o_test_utils.isVerbose(): print 'Done trying to build DeepLearning model with bad parameters.'
    
    #####################################
    # Early test of predict()
    # TODO: remove after we remove the early exit
    p = a_node.predict(model='deeplearning_airlines_binomial', frame='airlines_binomial', predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_airlines_binomial', 'airlines_binomial', 43978, predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_frame_exists(a_node, 'deeplearning_airlines_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_airlines_binomial', " on: ", 'airlines_binomial', ":  ", repr(p))
def test(a_node, pp):
    if h2o_test_utils.isVerbose():
        print "Testing ModelMetrics. . ."
    ###################################
    # Compute and check ModelMetrics for 'deeplearning_prostate_binomial'
    mm = a_node.compute_model_metrics(model="deeplearning_prostate_binomial", frame="prostate_binomial")
    assert mm is not None, (
        "FAIL: Got a null result for scoring: " + "deeplearning_prostate_binomial" + " on: " + "prostate_binomial"
    )
    assert "model_category" in mm, (
        "FAIL: ModelMetrics for scoring: "
        + "deeplearning_prostate_binomial"
        + " on: "
        + "prostate_binomial"
        + " does not contain a model_category."
    )
    assert "Binomial" == mm["model_category"], (
        "FAIL: ModelMetrics for scoring: "
        + "deeplearning_prostate_binomial"
        + " on: "
        + "prostate_binomial"
        + " model_category is not Binomial, it is: "
        + str(mm["model_category"])
    )
    assert "AUC" in mm, (
        "FAIL: ModelMetrics for scoring: "
        + "deeplearning_prostate_binomial"
        + " on: "
        + "prostate_binomial"
        + " does not contain an AUC element: "
        + h2o_test_utils.dump_json(mm)
    )
    assert type(mm["AUC"]) is float, (
        "FAIL: ModelMetrics for scoring: "
        + "deeplearning_prostate_binomial"
        + " on: "
        + "prostate_binomial"
        + " AUC element is not a float: "
        + h2o_test_utils.dump_json(mm)
    )

    assert "thresholds_and_metric_scores" in mm, (
        "FAIL: ModelMetrics for scoring: "
        + "deeplearning_prostate_binomial"
        + " on: "
        + "prostate_binomial"
        + " does not contain a thresholds_and_metric_scores element: "
        + h2o_test_utils.dump_json(mm)
    )
    assert type(mm["thresholds_and_metric_scores"]) is dict, (
        "FAIL: ModelMetrics for scoring: "
        + "deeplearning_prostate_binomial"
        + " on: "
        + "prostate_binomial"
        + " confusion_matrices element is not a dict: "
        + h2o_test_utils.dump_json(mm)
    )

    # print h2o_test_utils.dump_json(mm)
    h2o.H2O.verboseprint(
        "ModelMetrics for scoring: ", "deeplearning_prostate_binomial", " on: ", "prostate_binomial", ":  ", repr(mm)
    )

    ###################################
    # Check for ModelMetrics for 'deeplearning_prostate_binomial' in full list
    mms = a_node.model_metrics()  # fetch all
    assert "model_metrics" in mms, "FAIL: Failed to find model_metrics in result of /3/ModelMetrics."
    found_mm = False
    found_a_frame = False
    for mm in mms["model_metrics"]:
        assert "model" in mm, "FAIL: mm does not contain a model element: " + repr(mm)
        assert "name" in mm["model"], "FAIL: mm[model] isn't a key with a name: " + repr(mm)
        assert "type" in mm["model"], "FAIL: mm[model] does not contain a type: " + repr(mm)
        assert "Key<Model>" == mm["model"]["type"], "FAIL: mm[model] type is not Key<Model>: " + repr(
            mm["model"]["type"]
        )

        assert "frame" in mm, "FAIL: mm does not contain a frame element: " + repr(mm)

        if mm["frame"] is not None:  # temp metrics can have null frames
            assert "name" in mm["frame"], "FAIL: mm[frame] does not contain a name: " + repr(mm)
            assert "type" in mm["frame"], "FAIL: mm[frame] does not contain a type: " + repr(mm)
            assert "Key<Frame>" == mm["frame"]["type"], "FAIL: mm[frame] type is not Key<Frame>: " + repr(mm)

            model_key = mm["model"]["name"]
            frame_key = mm["frame"]["name"]  # TODO: should match
            if model_key == "deeplearning_prostate_binomial" and frame_key == "prostate_binomial":
                found_mm = True
        else:
            found_a_frame = True
    assert found_mm, (
        "FAIL: Failed to find ModelMetrics object for model: "
        + "deeplearning_prostate_binomial"
        + " and frame: "
        + "prostate_binomial"
    )
    assert found_a_frame, "FAIL: Failed to find a frame in any ModelMetrics object."

    # test delete_model_metrics
    mms = a_node.model_metrics("deeplearning_prostate_binomial", "prostate_binomial")
    assert len(mms["model_metrics"]) == 1, "FAIL: expected 1 ModelMetrics, found: " + str(len(mms["model_metrics"]))
    a_node.delete_model_metrics("deeplearning_prostate_binomial", "prostate_binomial")
    mms = a_node.model_metrics("deeplearning_prostate_binomial", "prostate_binomial")
    assert len(mms["model_metrics"]) == 0, "FAIL: expected 0 ModelMetrics, found: " + str(len(mms["model_metrics"]))

    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_binomial'
    p = a_node.predict(
        model="deeplearning_prostate_binomial",
        frame="prostate_binomial",
        predictions_frame="deeplearning_prostate_binomial_predictions",
    )
    h2o_test_utils.validate_predictions(
        a_node,
        p,
        "deeplearning_prostate_binomial",
        "prostate_binomial",
        380,
        predictions_frame="deeplearning_prostate_binomial_predictions",
    )
    h2o_test_utils.validate_frame_exists(a_node, "deeplearning_prostate_binomial_predictions")
    h2o.H2O.verboseprint(
        "Predictions for scoring: ", "deeplearning_prostate_binomial", " on: ", "prostate_binomial", ":  ", repr(p)
    )

    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_binomial'
    p = a_node.predict(model="gbm_prostate_binomial", frame="prostate_binomial")
    h2o_test_utils.validate_predictions(a_node, p, "gbm_prostate_binomial", "prostate_binomial", 380)
    h2o.H2O.verboseprint(
        "Predictions for scoring: ", "gbm_prostate_binomial", " on: ", "prostate_binomial", ":  ", repr(p)
    )

    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_regression'
    p = a_node.predict(model="deeplearning_prostate_regression", frame="prostate_regression")
    h2o_test_utils.validate_predictions(a_node, p, "deeplearning_prostate_regression", "prostate_regression", 380)
    h2o.H2O.verboseprint(
        "Predictions for scoring: ", "deeplearning_prostate_regression", " on: ", "prostate_regression", ":  ", repr(p)
    )

    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_regression'
    p = a_node.predict(model="gbm_prostate_regression", frame="prostate_regression")
    h2o_test_utils.validate_predictions(a_node, p, "gbm_prostate_regression", "prostate_regression", 380)
    h2o.H2O.verboseprint(
        "Predictions for scoring: ", "gbm_prostate_regression", " on: ", "prostate_regression", ":  ", repr(p)
    )

    ###################################
    # Predict and check ModelMetrics (empty now except for predictions frame) for 'kmeans_prostate'
    p = a_node.predict(model="kmeans_prostate", frame="prostate_regression")
    h2o_test_utils.validate_predictions(a_node, p, "kmeans_prostate", "prostate_regression", 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", "kmeans_prostate", " on: ", "prostate_regression", ":  ", repr(p))
def test(a_node, pp):
    ###################################
    # Compute and check ModelMetrics for 'deeplearning_prostate_binomial'
    mm = a_node.compute_model_metrics(model='deeplearning_prostate_binomial',
                                      frame='prostate_binomial')
    assert mm is not None, "FAIL: Got a null result for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial'
    assert 'model_category' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain a model_category."
    assert 'Binomial' == mm[
        'model_category'], "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " model_category is not Binomial, it is: " + str(
            mm['model_category'])
    assert 'AUC' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain an AUC element: " + h2o_test_utils.dump_json(
        mm)
    assert type(
        mm['AUC']
    ) is float, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " AUC element is not a float: " + h2o_test_utils.dump_json(
        mm)

    assert 'confusion_matrices' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain a confusion_matrices element: " + h2o_test_utils.dump_json(
        mm)
    assert type(
        mm['confusion_matrices']
    ) is list, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " confusion_matrices element is not a list: " + h2o_test_utils.dump_json(
        mm)

    # print h2o_test_utils.dump_json(mm)
    h2o.H2O.verboseprint("ModelMetrics for scoring: ",
                         'deeplearning_prostate_binomial', " on: ",
                         'prostate_binomial', ":  ", repr(mm))

    ###################################
    # Check for ModelMetrics for 'deeplearning_prostate_binomial' in full list
    mms = a_node.model_metrics()  # fetch all
    assert 'model_metrics' in mms, 'FAIL: Failed to find model_metrics in result of /3/ModelMetrics.'
    found_mm = False
    for mm in mms['model_metrics']:
        assert 'model' in mm, "FAIL: mm does not contain a model element: " + repr(
            mm)
        assert 'name' in mm[
            'model'], "FAIL: mm[model] isn't a key with a name: " + repr(mm)
        assert 'type' in mm[
            'model'], "FAIL: mm[model] does not contain a type: " + repr(mm)
        assert 'Key<Model>' == mm['model'][
            'type'], "FAIL: mm[model] type is not Key<Model>: " + repr(
                mm['model']['type'])

        assert 'frame' in mm, "FAIL: mm does not contain a frame element: " + repr(
            mm)
        assert 'name' in mm[
            'frame'], "FAIL: mm[frame] does not contain a name: " + repr(mm)
        assert 'type' in mm[
            'frame'], "FAIL: mm[frame] does not contain a type: " + repr(mm)
        assert 'Key<Frame>' == mm['frame'][
            'type'], "FAIL: mm[frame] type is not Key<Frame>: " + repr(mm)

        model_key = mm['model']['name']
        frame_key = mm['frame']['name']  # TODO: should match
        if model_key == 'deeplearning_prostate_binomial' and frame_key == 'prostate_binomial':
            found_mm = True
    assert found_mm, "FAIL: Failed to find ModelMetrics object for model: " + 'deeplearning_prostate_binomial' + " and frame: " + 'prostate_binomial'

    # test delete_model_metrics
    mms = a_node.model_metrics('deeplearning_prostate_binomial',
                               'prostate_binomial')
    assert len(mms['model_metrics']
               ) == 1, "FAIL: expected 1 ModelMetrics, found: " + str(
                   len(mms['model_metrics']))
    a_node.delete_model_metrics('deeplearning_prostate_binomial',
                                'prostate_binomial')
    mms = a_node.model_metrics('deeplearning_prostate_binomial',
                               'prostate_binomial')
    assert len(mms['model_metrics']
               ) == 0, "FAIL: expected 0 ModelMetrics, found: " + str(
                   len(mms['model_metrics']))

    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_binomial'
    p = a_node.predict(
        model='deeplearning_prostate_binomial',
        frame='prostate_binomial',
        predictions_frame='deeplearning_prostate_binomial_predictions')
    h2o_test_utils.validate_predictions(
        a_node,
        p,
        'deeplearning_prostate_binomial',
        'prostate_binomial',
        380,
        predictions_frame='deeplearning_prostate_binomial_predictions')
    h2o_test_utils.validate_frame_exists(
        a_node, 'deeplearning_prostate_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ",
                         'deeplearning_prostate_binomial', " on: ",
                         'prostate_binomial', ":  ", repr(p))

    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_regression'
    p = a_node.predict(model='deeplearning_prostate_regression',
                       frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p,
                                        'deeplearning_prostate_regression',
                                        'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ",
                         'deeplearning_prostate_regression', " on: ",
                         'prostate_binomial', ":  ", repr(p))

    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_binomial'
    p = a_node.predict(model='gbm_prostate_binomial',
                       frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'gbm_prostate_binomial',
                                        'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'gbm_prostate_binomial',
                         " on: ", 'prostate_binomial', ":  ", repr(p))

    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_regression'
    p = a_node.predict(model='gbm_prostate_regression',
                       frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'gbm_prostate_regression',
                                        'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ",
                         'gbm_prostate_regression', " on: ",
                         'prostate_binomial', ":  ", repr(p))

    ###################################
    # Predict and check ModelMetrics (empty now except for predictions frame) for 'kmeans_prostate'
    p = a_node.predict(model='kmeans_prostate', frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'kmeans_prostate',
                                        'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'kmeans_prostate',
                         " on: ", 'prostate_binomial', ":  ", repr(p))
Ejemplo n.º 6
0
def build_and_test(a_node, pp, datasets, algos,
                   algo_additional_default_params):
    ####################################################################################################
    # Build and do basic validation checks on models
    ####################################################################################################
    models_to_build = [
        ModelSpec.for_dataset('kmeans_prostate', 'kmeans',
                              datasets['prostate_clustering'], {'k': 2}),
        ModelSpec.for_dataset('glm_prostate_regression', 'glm',
                              datasets['prostate_regression'],
                              {'family': 'gaussian'}),
        ModelSpec.for_dataset('glm_prostate_binomial', 'glm',
                              datasets['prostate_binomial'],
                              {'family': 'binomial'}),
        ModelSpec.for_dataset('glm_airlines_binomial', 'glm',
                              datasets['airlines_binomial'], {
                                  'response_column': 'IsDepDelayed',
                                  'family': 'binomial'
                              }),
        ModelSpec.for_dataset('glm_iris_multinomial', 'glm',
                              datasets['iris_multinomial'], {
                                  'response_column': 'class',
                                  'family': 'multinomial'
                              }),
        ModelSpec.for_dataset('deeplearning_prostate_regression',
                              'deeplearning', datasets['prostate_regression'],
                              {
                                  'epochs': 1,
                                  'loss': 'Quadratic'
                              }),
        ModelSpec.for_dataset('deeplearning_prostate_binomial', 'deeplearning',
                              datasets['prostate_binomial'], {
                                  'epochs': 1,
                                  'hidden': [20, 20],
                                  'loss': 'CrossEntropy'
                              }),
        ModelSpec.for_dataset('deeplearning_airlines_binomial', 'deeplearning',
                              datasets['airlines_binomial'], {
                                  'epochs': 1,
                                  'hidden': [10, 10],
                                  'loss': 'CrossEntropy'
                              }),
        ModelSpec.for_dataset('deeplearning_iris_multinomial', 'deeplearning',
                              datasets['iris_multinomial'], {
                                  'epochs': 1,
                                  'loss': 'CrossEntropy'
                              }),
        ModelSpec.for_dataset('gbm_prostate_regression', 'gbm',
                              datasets['prostate_regression'], {
                                  'ntrees': 5,
                                  'distribution': 'gaussian'
                              }),
        ModelSpec.for_dataset('gbm_prostate_binomial', 'gbm',
                              datasets['prostate_binomial'], {
                                  'ntrees': 5,
                                  'distribution': 'multinomial'
                              }),
        ModelSpec.for_dataset('gbm_airlines_binomial', 'gbm',
                              datasets['airlines_binomial'], {
                                  'ntrees': 5,
                                  'distribution': 'multinomial'
                              }),
        ModelSpec.for_dataset('gbm_iris_multinomial', 'gbm',
                              datasets['iris_multinomial'], {
                                  'ntrees': 5,
                                  'distribution': 'multinomial'
                              }),
    ]

    # For grid testing, don't build any non-grid models:
    # models_to_build = []

    built_models = {}
    for model_spec in models_to_build:
        model = model_spec.build_and_validate_model(a_node)
        built_models[model_spec['dest_key']] = model

    grids_to_build = [
        # setting a hyperparameter in both places:
        # GridSpec.for_dataset('kmeans_prostate_grid', 'kmeans', datasets['prostate_clustering'], { 'k': 6 }, { 'k': [2, 3, 4] } ),
        GridSpec.for_dataset('kmeans_prostate_grid', 'kmeans',
                             datasets['prostate_clustering'], {},
                             {'k': [2, 3, 4]}),
        GridSpec.for_dataset('glm_prostate_regression_grid', 'glm',
                             datasets['prostate_regression'],
                             {'family': 'gaussian'},
                             {'lambda': [0.0001, 0.001, 0.01, 0.1]}),
        GridSpec.for_dataset('glm_prostate_binomial_grid', 'glm',
                             datasets['prostate_binomial'],
                             {'family': 'binomial'},
                             {'lambda': [0.0001, 0.001, 0.01, 0.1]}),
        GridSpec.for_dataset('glm_airlines_binomial_grid', 'glm',
                             datasets['airlines_binomial'], {
                                 'response_column': 'IsDepDelayed',
                                 'family': 'binomial'
                             }, {'lambda': [0.0001, 0.001, 0.01, 0.025]}),
        GridSpec.for_dataset('glm_iris_multinomial_grid', 'glm',
                             datasets['iris_multinomial'], {
                                 'response_column': 'class',
                                 'family': 'multinomial'
                             }, {'lambda': [0.0001, 0.001, 0.01, 0.025]}),
        GridSpec.for_dataset('deeplearning_prostate_regression_grid',
                             'deeplearning', datasets['prostate_regression'],
                             {'loss': 'Quadratic'}, {'epochs': [0.1, 0.5, 1]}),
        GridSpec.for_dataset('deeplearning_prostate_binomial_grid',
                             'deeplearning', datasets['prostate_binomial'], {
                                 'hidden': [20, 20],
                                 'loss': 'CrossEntropy'
                             }, {'epochs': [0.1, 0.5, 1]}),
        GridSpec.for_dataset('deeplearning_airlines_binomial_grid',
                             'deeplearning', datasets['airlines_binomial'], {
                                 'hidden': [10, 10],
                                 'loss': 'CrossEntropy'
                             }, {'epochs': [0.1, 0.5, 1]}),
        GridSpec.for_dataset('deeplearning_iris_multinomial_grid',
                             'deeplearning', datasets['iris_multinomial'],
                             {'loss': 'CrossEntropy'},
                             {'epochs': [0.1, 0.5, 1]}),
        GridSpec.for_dataset(
            'gbm_prostate_regression_grid', 'gbm',
            datasets['prostate_regression'], {'max_depth': 3}, {
                'ntrees': [1, 5, 10],
                'distribution': ["gaussian", "poisson", "gamma", "tweedie"]
            }),
        GridSpec.for_dataset('gbm_prostate_binomial_grid', 'gbm',
                             datasets['prostate_binomial'], {}, {
                                 'ntrees': [5, 7],
                                 'max_depth': [1, 3, 5]
                             }),
        GridSpec.for_dataset('gbm_airlines_binomial_grid', 'gbm',
                             datasets['airlines_binomial'],
                             {'distribution': 'multinomial'}, {
                                 'ntrees': [1, 5, 10],
                                 'max_depth': [1, 3, 5]
                             }),
        GridSpec.for_dataset('gbm_iris_multinomial_grid', 'gbm',
                             datasets['iris_multinomial'],
                             {'distribution': 'multinomial'}, {
                                 'ntrees': [1, 5, 10],
                                 'max_depth': [1, 3, 5]
                             }),
        # TODO: this should trigger a parameter validation error, but instead the non-grid ntrees silently overrides the grid values:        GridSpec.for_dataset('gbm_iris_multinomial_grid', 'gbm', datasets['iris_multinomial'], { 'ntrees': 5, 'distribution': 'multinomial' }, { 'ntrees': [1, 5, 10], 'max_depth': [1, 3, 5] } ),

        # Test stopping criteria:
        GridSpec.for_dataset(
            'gbm_prostate_regression_grid_max_3', 'gbm',
            datasets['prostate_regression'], {'max_depth': 3}, {
                'ntrees': [1, 2, 4],
                'distribution': ["gaussian", "poisson", "gamma", "tweedie"]
            }, {
                'strategy': "RandomDiscrete",
                'max_models': 3
            }),
        GridSpec.for_dataset(
            'gbm_prostate_regression_grid_max_20mS', 'gbm',
            datasets['prostate_regression'], {'max_depth': 3}, {
                'ntrees': [1, 2, 4],
                'distribution': ["gaussian", "poisson", "gamma", "tweedie"]
            }, {
                'strategy': "RandomDiscrete",
                'max_runtime_secs': 0.020
            }),
        GridSpec.for_dataset(
            'gbm_prostate_regression_grid_stopping_deviance', 'gbm',
            datasets['prostate_regression'], {}, {
                'max_depth': [1, 2, 3, 4, 5, 6, 7],
                'ntrees': [1, 2, 3, 4, 5, 6],
                'distribution': ["gaussian", "poisson", "gamma"]
            }, {
                'strategy': "RandomDiscrete",
                'seed': 42,
                'stopping_metric': 'deviance',
                'stopping_tolerance': 0.00001,
                'stopping_rounds': 5
            }),
        GridSpec.for_dataset(
            'gbm_prostate_regression_grid_stopping_auto', 'gbm',
            datasets['prostate_regression'], {}, {
                'max_depth': [1, 2, 3, 4, 5, 6, 7],
                'ntrees': [1, 2, 3, 4, 5, 6],
                'distribution': ["gaussian", "poisson", "gamma"]
            }, {
                'strategy': "RandomDiscrete",
                'seed': 42,
                'stopping_metric': 'AUTO',
                'stopping_tolerance': 0.00001,
                'stopping_rounds': 5
            }),
    ]

    for grid_spec in grids_to_build:
        grid = grid_spec.build_and_validate_grid(a_node)

        for model_key in grid['model_ids']:
            model_key = model_key['name']
            built_models[model_key] = a_node.models(key=model_key)

    # test search limits: max_models
    grid = a_node.grid(key='gbm_prostate_regression_grid_max_3')
    assert len(
        grid['model_ids']
    ) == 3, "FAIL: using max_models, expected a max of 3 models, got: " + str(
        len(grid['model_ids']))

    # test search limits: max_runtime_secs
    grid = a_node.grid(key='gbm_prostate_regression_grid_max_20mS')
    assert len(
        grid['model_ids']
    ) < 12, "FAIL: using max_runtime_secs, expected less than 12 models, got: " + str(
        len(grid['model_ids']))

    # test search limits: stopping_deviance
    grid = a_node.grid(key='gbm_prostate_regression_grid_stopping_deviance')
    deviance_model_count = len(grid['model_ids'])
    assert len(
        grid['model_ids']
    ) < 126, "FAIL: using asymptotic deviance stopping criterion, expected less than 126 models, got: " + str(
        len(grid['model_ids']))

    # test search limits: stopping_auto
    grid = a_node.grid(key='gbm_prostate_regression_grid_stopping_auto')
    auto_model_count = len(grid['model_ids'])
    assert len(
        grid['model_ids']
    ) < 126, "FAIL: using asymptotic auto stopping criterion, expected less than 126 models, got: " + str(
        len(grid['model_ids']))

    # test that AUTO gave the same answer as deviance
    assert deviance_model_count == auto_model_count, "FAIL: using asymptotic auto stopping criterion, expected the same number of models as deviance, got: " + str(
        auto_model_count) + " instead of: " + str(deviance_model_count)

    #    grid = a_node.grid(key='kmeans_prostate_grid', sort_by='', decreasing=True)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node,
                                                key='kmeans_prostate_grid',
                                                sort_by='totss',
                                                decreasing=True)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node,
                                                key='kmeans_prostate_grid',
                                                sort_by='tot_withinss',
                                                decreasing=True)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node,
                                                key='kmeans_prostate_grid',
                                                sort_by='betweenss',
                                                decreasing=True)

    h2o_test_utils.fetch_and_validate_grid_sort(a_node,
                                                key='kmeans_prostate_grid',
                                                sort_by='totss',
                                                decreasing=False)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node,
                                                key='kmeans_prostate_grid',
                                                sort_by='tot_withinss',
                                                decreasing=False)
    h2o_test_utils.fetch_and_validate_grid_sort(a_node,
                                                key='kmeans_prostate_grid',
                                                sort_by='betweenss',
                                                decreasing=False)

    #    import sys
    #    sys.exit(0)
    #######################################
    # Test default parameters validation for each model builder
    #
    if h2o_test_utils.isVerbose():
        print('Testing ModelBuilder default parameters. . .')
    model_builders = a_node.model_builders(timeoutSecs=240)['model_builders']

    # Do we know about all of them?
    server_algos = model_builders.keys()
    assert len(
        set(server_algos) - set(algos)
    ) == 0, "FAIL: Our set of algos doesn't match what the server knows about.  Ours: " + repr(
        algos) + "; server's: " + repr(server_algos)

    for algo, model_builder in model_builders.iteritems():
        parameters_list = model_builder['parameters']
        test_parameters = {
            value['name']: value['default_value']
            for value in parameters_list
        }  # collect default parameters
        if algo in algo_additional_default_params:
            test_parameters.update(algo_additional_default_params[algo])

        if h2o_test_utils.isVerboser():
            print('Testing ' + algo + ' with params: ' + repr(test_parameters))

        parameters_validation = a_node.validate_model_parameters(
            algo=algo,
            training_frame=None,
            parameters=test_parameters,
            timeoutSecs=240)  # synchronous
        assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
        h2o.H2O.verboseprint("Bad params validation messages: ",
                             repr(parameters_validation))

        expected_count = 0
        if expected_count != parameters_validation['error_count']:
            print("validation errors: ")
            pp.pprint(parameters_validation)
        assert expected_count == parameters_validation[
            'error_count'], "FAIL: " + str(
                expected_count
            ) + " != error_count in good-parameters parameters validation result."

    #######################################
    # Test DeepLearning parameters validation
    #
    # Default parameters:
    if h2o_test_utils.isVerbose():
        print('Testing DeepLearning default parameters. . .')
    model_builder = a_node.model_builders(
        algo='deeplearning', timeoutSecs=240)['model_builders']['deeplearning']
    dl_test_parameters_list = model_builder['parameters']
    dl_test_parameters = {
        value['name']: value['default_value']
        for value in dl_test_parameters_list
    }

    parameters_validation = a_node.validate_model_parameters(
        algo='deeplearning',
        training_frame=None,
        parameters=dl_test_parameters,
        timeoutSecs=240)  # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ",
                         repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print("validation errors: ")
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation[
        'error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."

    # Good parameters (note: testing with null training_frame):
    if h2o_test_utils.isVerbose():
        print('Testing DeepLearning good parameters. . .')
    dl_test_parameters = {
        'response_column': 'CAPSULE',
        'hidden': "[10, 20, 10]"
    }
    parameters_validation = a_node.validate_model_parameters(
        algo='deeplearning',
        training_frame=None,
        parameters=dl_test_parameters,
        timeoutSecs=240)  # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in good-parameters parameters validation result."
    h2o.H2O.verboseprint("Bad params validation messages: ",
                         repr(parameters_validation))
    if 0 != parameters_validation['error_count']:
        print("validation errors: ")
        pp.pprint(parameters_validation)
    assert 0 == parameters_validation[
        'error_count'], "FAIL: 0 != error_count in good-parameters parameters validation result."

    # Bad parameters (hidden is null):
    # (note: testing with null training_frame)
    if h2o_test_utils.isVerbose():
        print('Testing DeepLearning bad parameters, null training_frame. . .')
    dl_test_parameters = {
        'response_column': 'CAPSULE',
        'hidden': "[10, 20, 10]",
        'input_dropout_ratio': 27
    }
    parameters_validation = a_node.validate_model_parameters(
        algo='deeplearning',
        training_frame=None,
        parameters=dl_test_parameters,
        timeoutSecs=240)  # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (input_dropout_ratio)."
    h2o.H2O.verboseprint("Good params validation messages: ",
                         repr(parameters_validation))
    assert 0 != parameters_validation[
        'error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(
            parameters_validation)

    found_expected_error = False
    for validation_message in parameters_validation['messages']:
        if validation_message['message_type'] == 'ERRR' and validation_message[
                'field_name'] == 'input_dropout_ratio':
            found_expected_error = True
    assert found_expected_error, "FAIL: Failed to find error message about input_dropout_ratio in the validation messages."

    # Bad parameters (no response_column):
    if h2o_test_utils.isVerbose():
        print('Testing DeepLearning bad parameters, null response_column. . .')
    dl_test_parameters = {'hidden': "[10, 20, 10]"}
    parameters_validation = a_node.validate_model_parameters(
        algo='deeplearning',
        training_frame='prostate_binomial',
        parameters=dl_test_parameters,
        timeoutSecs=240)  # synchronous
    assert 'error_count' in parameters_validation, "FAIL: Failed to find error_count in bad-parameters parameters validation result (response_column)."
    h2o.H2O.verboseprint("Good params validation messages: ",
                         repr(parameters_validation))
    assert 0 != parameters_validation[
        'error_count'], "FAIL: 0 == error_count in bad-parameters parameters validation result: " + repr(
            parameters_validation)

    #######################################
    # Try to build DeepLearning model for Prostate but with bad parameters; we should get a ModelParametersSchema with the error.
    if h2o_test_utils.isVerbose():
        print(
            'About to try to build a DeepLearning model with bad parameters. . .'
        )
    dl_prostate_bad_parameters = {
        'response_column': 'CAPSULE',
        'hidden': "[10, 20, 10]",
        'input_dropout_ratio': 27
    }
    parameters_validation = a_node.build_model(
        algo='deeplearning',
        model_id='deeplearning_prostate_binomial_bad',
        training_frame='prostate_binomial',
        parameters=dl_prostate_bad_parameters,
        timeoutSecs=240)  # synchronous
    h2o_test_utils.validate_validation_messages(parameters_validation,
                                                ['input_dropout_ratio'])
    assert parameters_validation['__http_response'][
        'status_code'] == requests.codes.precondition_failed, "FAIL: expected 412 Precondition Failed from a bad build request, got: " + str(
            parameters_validation['__http_response']['status_code'])
    if h2o_test_utils.isVerbose():
        print('Done trying to build DeepLearning model with bad parameters.')

    #####################################
    # Early test of predict()
    # TODO: remove after we remove the early exit
    p = a_node.predict(
        model='deeplearning_airlines_binomial',
        frame='airlines_binomial',
        predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_predictions(
        a_node,
        p,
        'deeplearning_airlines_binomial',
        'airlines_binomial',
        43978,
        predictions_frame='deeplearning_airlines_binomial_predictions')
    h2o_test_utils.validate_frame_exists(
        a_node, 'deeplearning_airlines_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ",
                         'deeplearning_airlines_binomial', " on: ",
                         'airlines_binomial', ":  ", repr(p))
def test(a_node, pp):
    ###################################
    # Compute and check ModelMetrics for 'deeplearning_prostate_binomial'
    mm = a_node.compute_model_metrics(model='deeplearning_prostate_binomial', frame='prostate_binomial')
    assert mm is not None, "FAIL: Got a null result for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial'
    assert 'model_category' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain a model_category."
    assert 'Binomial' == mm['model_category'], "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " model_category is not Binomial, it is: " + str(mm['model_category'])
    assert 'AUC' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain an AUC element: " + h2o_test_utils.dump_json(mm)
    assert type(mm['AUC']) is float, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " AUC element is not a float: " + h2o_test_utils.dump_json(mm)
    
    assert 'confusion_matrices' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain a confusion_matrices element: " + h2o_test_utils.dump_json(mm)
    assert type(mm['confusion_matrices']) is list, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " confusion_matrices element is not a list: " + h2o_test_utils.dump_json(mm)
    
    # print h2o_test_utils.dump_json(mm)
    h2o.H2O.verboseprint("ModelMetrics for scoring: ", 'deeplearning_prostate_binomial', " on: ", 'prostate_binomial', ":  ", repr(mm))
    
    ###################################
    # Check for ModelMetrics for 'deeplearning_prostate_binomial' in full list
    mms = a_node.model_metrics() # fetch all
    assert 'model_metrics' in mms, 'FAIL: Failed to find model_metrics in result of /3/ModelMetrics.'
    found_mm = False
    for mm in mms['model_metrics']:
        assert 'model' in mm, "FAIL: mm does not contain a model element: " + repr(mm)
        assert 'name' in mm['model'], "FAIL: mm[model] isn't a key with a name: " + repr(mm)
        assert 'type' in mm['model'], "FAIL: mm[model] does not contain a type: " + repr(mm)
        assert 'Key<Model>' == mm['model']['type'], "FAIL: mm[model] type is not Key<Model>: " + repr(mm['model']['type'])
    
        assert 'frame' in mm, "FAIL: mm does not contain a frame element: " + repr(mm)
        assert 'name' in mm['frame'], "FAIL: mm[frame] does not contain a name: " + repr(mm)
        assert 'type' in mm['frame'], "FAIL: mm[frame] does not contain a type: " + repr(mm)
        assert 'Key<Frame>' == mm['frame']['type'], "FAIL: mm[frame] type is not Key<Frame>: " + repr(mm)
    
        model_key = mm['model']['name']
        frame_key = mm['frame']['name'] # TODO: should match
        if model_key == 'deeplearning_prostate_binomial' and frame_key == 'prostate_binomial':
            found_mm = True
    assert found_mm, "FAIL: Failed to find ModelMetrics object for model: " + 'deeplearning_prostate_binomial' + " and frame: " + 'prostate_binomial'
    
    # test delete_model_metrics
    mms = a_node.model_metrics('deeplearning_prostate_binomial', 'prostate_binomial')
    assert len(mms['model_metrics']) == 1, "FAIL: expected 1 ModelMetrics, found: " + str(len(mms['model_metrics']))
    a_node.delete_model_metrics('deeplearning_prostate_binomial', 'prostate_binomial')
    mms = a_node.model_metrics('deeplearning_prostate_binomial', 'prostate_binomial')
    assert len(mms['model_metrics']) == 0, "FAIL: expected 0 ModelMetrics, found: " + str(len(mms['model_metrics']))
    
    
    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_binomial'
    p = a_node.predict(model='deeplearning_prostate_binomial', frame='prostate_binomial', predictions_frame='deeplearning_prostate_binomial_predictions')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_prostate_binomial', 'prostate_binomial', 380, predictions_frame='deeplearning_prostate_binomial_predictions')
    h2o_test_utils.validate_frame_exists(a_node, 'deeplearning_prostate_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_prostate_binomial', " on: ", 'prostate_binomial', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_regression'
    p = a_node.predict(model='deeplearning_prostate_regression', frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_prostate_regression', 'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_prostate_regression', " on: ", 'prostate_binomial', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_binomial'
    p = a_node.predict(model='gbm_prostate_binomial', frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'gbm_prostate_binomial', 'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'gbm_prostate_binomial', " on: ", 'prostate_binomial', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_regression'
    p = a_node.predict(model='gbm_prostate_regression', frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'gbm_prostate_regression', 'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'gbm_prostate_regression', " on: ", 'prostate_binomial', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics (empty now except for predictions frame) for 'kmeans_prostate'
    p = a_node.predict(model='kmeans_prostate', frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'kmeans_prostate', 'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'kmeans_prostate', " on: ", 'prostate_binomial', ":  ", repr(p))
Ejemplo n.º 8
0
def test(a_node, pp):
    if h2o_test_utils.isVerbose(): print('Testing ModelMetrics. . .')
    ###################################
    # Compute and check ModelMetrics for 'deeplearning_prostate_binomial'
    mm = a_node.compute_model_metrics(model='deeplearning_prostate_binomial', frame='prostate_binomial')
    assert mm is not None, "FAIL: Got a null result for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial'
    assert 'model_category' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain a model_category."
    assert 'Binomial' == mm['model_category'], "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " model_category is not Binomial, it is: " + str(mm['model_category'])
    assert 'AUC' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain an AUC element: " + h2o_test_utils.dump_json(mm)
    assert type(mm['AUC']) is float, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " AUC element is not a float: " + h2o_test_utils.dump_json(mm)
    
    assert 'thresholds_and_metric_scores' in mm, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " does not contain a thresholds_and_metric_scores element: " + h2o_test_utils.dump_json(mm)
    assert type(mm['thresholds_and_metric_scores']) is dict, "FAIL: ModelMetrics for scoring: " + 'deeplearning_prostate_binomial' + " on: " + 'prostate_binomial' + " confusion_matrices element is not a dict: " + h2o_test_utils.dump_json(mm)
    
    # print(h2o_test_utils.dump_json(mm))
    h2o.H2O.verboseprint("ModelMetrics for scoring: ", 'deeplearning_prostate_binomial', " on: ", 'prostate_binomial', ":  ", repr(mm))
    
    ###################################
    # Check for ModelMetrics for 'deeplearning_prostate_binomial' in full list
    mms = a_node.model_metrics() # fetch all
    assert 'model_metrics' in mms, 'FAIL: Failed to find model_metrics in result of /3/ModelMetrics.'
    found_mm = False
    found_a_frame = False
    for mm in mms['model_metrics']:
        assert 'model' in mm, "FAIL: mm does not contain a model element: " + repr(mm)
        assert 'name' in mm['model'], "FAIL: mm[model] isn't a key with a name: " + repr(mm)
        assert 'type' in mm['model'], "FAIL: mm[model] does not contain a type: " + repr(mm)
        assert 'Key<Model>' == mm['model']['type'], "FAIL: mm[model] type is not Key<Model>: " + repr(mm['model']['type'])
    
        assert 'frame' in mm, "FAIL: mm does not contain a frame element: " + repr(mm)

        if mm['frame'] is not None:  # temp metrics can have null frames
            assert 'name' in mm['frame'], "FAIL: mm[frame] does not contain a name: " + repr(mm)
            assert 'type' in mm['frame'], "FAIL: mm[frame] does not contain a type: " + repr(mm)
            assert 'Key<Frame>' == mm['frame']['type'], "FAIL: mm[frame] type is not Key<Frame>: " + repr(mm)
    
            model_key = mm['model']['name']
            frame_key = mm['frame']['name']
            if model_key == 'deeplearning_prostate_binomial' and frame_key == 'prostate_binomial':
                found_mm = True
        else:
            found_a_frame = True
    assert found_mm, "FAIL: Failed to find ModelMetrics object for model: " + 'deeplearning_prostate_binomial' + " and frame: " + 'prostate_binomial'
    assert found_a_frame, "FAIL: Failed to find a frame in any ModelMetrics object."
    
    # test delete_model_metrics
    mms = a_node.model_metrics('deeplearning_prostate_binomial', 'prostate_binomial')
    assert len(mms['model_metrics']) == 1, "FAIL: expected 1 ModelMetrics, found: " + str(len(mms['model_metrics']))
    a_node.delete_model_metrics('deeplearning_prostate_binomial', 'prostate_binomial')
    mms = a_node.model_metrics('deeplearning_prostate_binomial', 'prostate_binomial')
    assert len(mms['model_metrics']) == 0, "FAIL: expected 0 ModelMetrics, found: " + str(len(mms['model_metrics']))
    
    
    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_binomial'
    p = a_node.predict(model='deeplearning_prostate_binomial', frame='prostate_binomial', predictions_frame='deeplearning_prostate_binomial_predictions')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_prostate_binomial', 'prostate_binomial', 380, predictions_frame='deeplearning_prostate_binomial_predictions')
    h2o_test_utils.validate_frame_exists(a_node, 'deeplearning_prostate_binomial_predictions')
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_prostate_binomial', " on: ", 'prostate_binomial', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_binomial'
    p = a_node.predict(model='gbm_prostate_binomial', frame='prostate_binomial')
    h2o_test_utils.validate_predictions(a_node, p, 'gbm_prostate_binomial', 'prostate_binomial', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'gbm_prostate_binomial', " on: ", 'prostate_binomial', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics for 'deeplearning_prostate_regression'
    p = a_node.predict(model='deeplearning_prostate_regression', frame='prostate_regression')
    h2o_test_utils.validate_predictions(a_node, p, 'deeplearning_prostate_regression', 'prostate_regression', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'deeplearning_prostate_regression', " on: ", 'prostate_regression', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics for 'gbm_prostate_regression'
    p = a_node.predict(model='gbm_prostate_regression', frame='prostate_regression')
    h2o_test_utils.validate_predictions(a_node, p, 'gbm_prostate_regression', 'prostate_regression', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'gbm_prostate_regression', " on: ", 'prostate_regression', ":  ", repr(p))
    
    ###################################
    # Predict and check ModelMetrics (empty now except for predictions frame) for 'kmeans_prostate'
    p = a_node.predict(model='kmeans_prostate', frame='prostate_regression')
    h2o_test_utils.validate_predictions(a_node, p, 'kmeans_prostate', 'prostate_regression', 380)
    h2o.H2O.verboseprint("Predictions for scoring: ", 'kmeans_prostate', " on: ", 'prostate_regression', ":  ", repr(p))
    
    ###################################
    # Predict with reversed keys (should get an H2OErrorV1):
    try:
        p = a_node.predict(frame='kmeans_prostate', model='prostate_regression')
        assert 1 == 0, "FAIL: expected a ValueError exception looking for Model 'prostate_regression'"
    except Exception as e:
        pass